mirror of https://github.com/WongKinYiu/yolov7.git
update cls readme |
||
---|---|---|
.. | ||
data | ||
models | ||
utils | ||
LICENSE | ||
README.md | ||
classifier.py | ||
detect.py | ||
export.py | ||
hubconf.py | ||
requirements.txt | ||
setup.cfg | ||
train.py | ||
val.py |
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
Top-1: 78.3%
Top-5: 94.1%
Top-1: 78.2%
Top-5: 94.3%