GPU 환경
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.32.03 Driver Version: 460.32.03 CUDA Version: 11.2 |
+---------------------+-------------------------+-----------------------------+
사용 환경
+-----------------------------------------------------------------------------+
| - torch=1.7.1+cu110 |
| - torchcontrib==0.0.2 |
| - torchvision==0.8.2+cu110 |
| - cudatoolkit=11.3.1 |
| - matplotlib |
| - numpy |
| - pillow=9.2.0 |
| - opencv-contrib-python==4.7.0 |
| - pandas |
| - |
| - |
+---------------------+-------------------------+-----------------------------+
해당 프로젝트에서 GPU 자원이 필요한 부분은 3가지이다. OpenCV에서 인식율 향상을 위해 Super Resolution 진행할때, 도면 내에서 심볼을 탐지하기 위해QATM(해당 프로젝트에서는 사전학습된 VGG19 모델을 사용하였다.) Vision model architecture 을 사용할때 마지막으로 도면 내 텍스트를 인식하는데 사용되는 EasyOCR GPU 자원을 활용가능해 해당 옵션을 사용하였다.