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 |