AlexeyAB / darknet

YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
http://pjreddie.com/darknet/
Other
21.78k stars 7.96k forks source link

How can I get the values of AP small and AP large of MS COCO? #7497

Open Philharmy-Wang opened 3 years ago

Philharmy-Wang commented 3 years ago

I have trained my model on the MS COCO dataset, and I know how to get the value of AP0.75. But the paper needs the values of AP small and AP large. I want to know how these values are obtained. Is there any command in darknet to get these values directly?

AlexeyAB commented 3 years ago

It seems that you just trained with COCO-train-2014 and tested with COCO-test-2017 or trained with COCO-val-2017 and tested with COCO-val-2014.

To get fair numbers just validate your model on Codalab-server on test-dev dataset as described there: https://github.com/AlexeyAB/darknet/wiki/How-to-evaluate-accuracy-and-speed-of-YOLOv4#how-to-evaluate-ap-of-yolov4-on-the-ms-coco-evaluation-server