yolov7/cls
Kin-Yiu, Wong e2b32bad4f
u7 code
update cls readme
2022-09-05 09:42:49 +08:00
..
data u7 code 2022-09-01 20:13:32 +08:00
models u7 code 2022-09-01 20:13:32 +08:00
utils u7 code 2022-09-01 20:13:32 +08:00
LICENSE u7 code 2022-09-01 20:13:32 +08:00
README.md u7 code 2022-09-05 09:42:49 +08:00
classifier.py u7 code 2022-09-01 20:13:32 +08:00
detect.py u7 code 2022-09-01 20:13:32 +08:00
export.py u7 code 2022-09-01 20:13:32 +08:00
hubconf.py u7 code 2022-09-01 20:13:32 +08:00
requirements.txt u7 code 2022-09-01 20:13:32 +08:00
setup.cfg u7 code 2022-09-01 20:13:32 +08:00
train.py u7 code 2022-09-01 20:13:32 +08:00
val.py u7 code 2022-09-01 20:13:32 +08:00

README.md

yolov7

Implementation of "YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors"

This implimentation is based on yolov5.

All of installation, data preparation, and usage are as same as yolov5.

Training

python classifier.py --pretrained False --data imagenet --epochs 90 --img 224 --batch 256 --model yolov7 --name yolov7-cls --lr0 0.1 --optimizer SGD

Results

yolov7-cls.pt

Top-1: 78.3%
Top-5: 94.1%

yolov7-clsn.pt

Top-1: 78.2%
Top-5: 94.3%