Complex YOLO setup

[1] github 사이트로 이동

github.com/ghimiredhikura/Complex-YOLOv3

ghimiredhikura/Complex-YOLOv3

[2] git clone 이후, requirement 설치

 

발생 가능한 에러1

sudo -H pip install --upgrade --ignore-installed pip setuptools

에러2

참고사이트 : financedata.github.io/posts/faq_pip_packge_fail.html

 

효능있는 조치]

$ sudo pip3 install -U pip
$ sudo pip3 install -U setuptools

에러3

pip install opencv-python

에러4

pip install shapely

pip install mayavi

---

Complex Yolov4 버전 사용 [github.com/maudzung/Complex-YOLOv4-Pytorch]

[1] 데이터 셋을 github 맨 밑에서 제시하는 경로에 모두 위치시켜준다.

[2] test.py 실행

[에러해결]

해당 명령어를 넣었을때, 위의 에러가 발생하여 test.py에서 "map_location" argument 를 추가시켜줬다.

[3] evaluation.py 실행

(test.py 에서 발생한 에러가 동일하게 발생하므로 evaluate.py 에서 "map_location" argument 를 추가시켜준다.)

$ python evaluate.py --gpu_idx 0 --pretrained_path ../checkpoints/complex_yolov4/complex_yolov4_mse_loss.pth --cfgfile ./config/cfg/complex_yolov4.cfg --img_size 608 --conf-thresh 0.5 --nms-thresh 0.5 --iou-thresh 0.5