Yolov7 based Image Classification

Machine Learning의 가장 기본 중의 기본인 Image classification을 Yolo를 가지고 해보자

 

사실 Yolov7에는 별도로 구현되어 있지 않다. 

 

때문에 들여다 보면 Yolov5 에서 사용되었던 image classification code를 그대로 가져다 사용하는 것을 알 수 있다.

 

관련 Repository는 다음과 같다

https://github.com/WongKinYiu/yolov7/tree/u7/det

GitHub - WongKinYiu/yolov7: Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time

방법은 늘 그랬듯이 그전 Yolo series와 동일하다

 

Prediction (Test) procedure

python classify/predict.py --weights yolov5s-cls.pt --source im.jpg

Training, Finetuning, Backbone change는 다음에 알아보자 (Absolutely!!)