Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Expected accuracy for YOLOv7-tiny leakyRelu #106

Open
atangfan opened this issue Jul 12, 2022 · 3 comments
Open

Expected accuracy for YOLOv7-tiny leakyRelu #106

atangfan opened this issue Jul 12, 2022 · 3 comments

Comments

@atangfan
Copy link

atangfan commented Jul 12, 2022

What is the expected accuracy for YOLOv7-tiny leakyRelu?
I got a very low AP @ 0.75 by using COCO2017-val

 Average Precision  (AP) @[ IoU=0.50      | area=   all | maxDets=100 ] = 0.458
 Average Precision  (AP) @[ IoU=0.75      | area=   all | maxDets=100 ] = 0.007

Should this be the same as SILU version?
Could you share experiments for YOLOv7 with leakyRelu if you have? Thanks!

@WongKinYiu
Copy link
Owner

YOLOv7-tiny
416: 35.2% AP, 52.8% AP50, 37.3% AP75.
320: 30.8% AP, 47.3% AP50, 32.2% AP75.

python train.py --workers 8 --device 0 --batch-size 64 --data data/coco.yaml --img 512 512 --cfg cfg/training/yolov7-tiny.yaml --weights '' --name yolov7-tiny --hyp data/hyp.scratch.tiny.yaml

@chjej202
Copy link

Hi, @WongKinYiu

what is the coco 2017 val accuracy with yolov7-tiny leaky?

I got the following mAP with your uploaded yolov7-tiny.pt (val2017)
python test.py --data data/coco.yaml --img 416 --batch 32 --conf 0.001 --iou 0.5 --device 0 --weights yolov7-tiny.pt --name yolov7tiny_416_val

Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.331
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.505
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.345
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.117
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.359
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.541
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.291
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.457
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.489
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.227
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.553
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.724

I also tested with the darknet with your uploaded yolov7-tiny.cfg and yolov7-tiny.weights (val2017)
./darknet detector map ./cfg/coco.data ./cfg/yolov7-tiny.cfg yolov7-tiny.weights
mean average precision (mAP@0.50) = 0.558828, or 55.88 %

I got different results from your yolov7-tiny.weights and yolov7-tiny.pt, and my results are not equal to your accuracy mentioned on the above comment.

what is the expected accuracy should I get?

@WongKinYiu
Copy link
Owner

The yolov7-tiny.pt is uploaded for someone who want to use it for onnx/tensorrt testing. That is the reason why yolov7-tiny is not included in readme of main branch.

We develop tiny models on darknet compatible branch, you could use yolor or pytorch yolov4 to get expected accuracy.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants