update cls readme
pull/841/head
Kin-Yiu, Wong 2022-09-05 09:42:49 +08:00 committed by GitHub
parent bcc3eff9e2
commit e2b32bad4f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 0 deletions

View File

@ -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%
```