mirror of https://github.com/WongKinYiu/yolov7.git
parent
bcc3eff9e2
commit
e2b32bad4f
cls
|
@ -11,3 +11,19 @@ All of installation, data preparation, and usage are as same as yolov5.
|
|||
``` shell
|
||||
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`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-cls.pt)
|
||||
|
||||
```
|
||||
Top-1: 78.3%
|
||||
Top-5: 94.1%
|
||||
```
|
||||
|
||||
[`yolov7-clsn.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-clsn.pt)
|
||||
|
||||
```
|
||||
Top-1: 78.2%
|
||||
Top-5: 94.3%
|
||||
```
|
||||
|
|
Loading…
Reference in New Issue