YoloAir based Object Detection

Backbone change 를 위하여 Official Yolov7 repository를 사용해보려했으나,

 

안타깝게도 변환이 쉽지는 않을 것 같다.. 

Official yolov7 쪽에 문의를 해보았고, 답변이 yolov7이 담겨있는 또 다른 repo인 YoloAir에서 backbone을 가져와서 써보라고 조언해줬지만 크게 구조를 바꾸지 않는 이상 쉽지 않다는 결론이 났다.

(예를 들어 module이 담겨있는 models.Models.common의 경우 기존에 있던 modules (CSP, ST2CSP 등) 들이 완전히 사라지고 다른 곳으로 옮겨져 있는 것 같다. 실제 이를 완벽하게 교정하기 위해서는 꽤나 오랜 시간이 걸릴 것으로 판단된다.)

 

---

이러한 이유로 YoloAir에서 다시금 Object detection을 위한 

1. Training + Test / Transfer Learning
2. Inference on Image and Video
3. Backbone change

를 수행해보자

 

Custom dataset를 다루는건 나중으로 미루자... 너무 많다...

---

Training

Official Yolov7과는 매우 비슷하며 여기서는 efficient

python train.py --workers 8 --device 0,1,2,3 --batch-size 32 --data data/coco.yaml --img 640 --cfg configs/backbone/yolov5_efficient.yaml --weights '' --name yolov5_efficient_training --hyp data/hyps/hyp.scratch-low.yaml