|
--- |
|
pipeline_tag: object-detection |
|
--- |
|
|
|
# Setup |
|
Clone repo YoloV7 classification |
|
``` |
|
git clone -b cls https://github.com/kngcotofure/yolov7-gpl.git |
|
``` |
|
|
|
Install requirements |
|
|
|
``` |
|
cd yolov7-gpl/det |
|
pip install -r requirements.txt -q |
|
``` |
|
|
|
# Inference |
|
``` |
|
python classify/predict.py --weights yolov7-base-150e.pt --device 0 --source sample.jpg |
|
``` |
|
|
|
# Accuracy |
|
FairFace dataset https://paperswithcode.com/sota/facial-attribute-classification-on-fairface |
|
| | Model size | Input size | Top 1 Acc | |
|
| ------------- | ------------- | ------------- | ------------- | |
|
| MiVOLO-V2 | - | 384 | 62.28 | |
|
| MiVOLO-V1 | 105MB | 224 | 61.07 | |
|
| FairFace | 81.3MB | 224 | 59.7 | |
|
| YoloV-tiny-150e (our) | 12MB | 224 | 58 | |
|
| YoloV-base-150e (our) | 58MB | 224 | 60 | |
