tiennv's picture
Update README.md
c00fa7d verified
metadata
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