search

cocodataset / cocoapi

COCO API - Dataset @ http://cocodataset.org/
Other
6.01k stars 3.76k forks source link

Structur of the variables (Precision, Recall) - Calculate and draw ROC-Curve (Precision-Recall-Curve) #448

Open johnny-mueller opened 3 years ago

johnny-mueller commented 3 years ago

I use pycocotools to evaluate the MS COCO 2017 data set. And I have seen that after the python .accumulate() function Precision and Recall are available.

Precision contains an array with a structure of (10, 101, 80, 4, 3) and Recall contains an array with a structure of (10, 80, 4, 3).

Currently I can't do much with the structure of the two variables but I would like to draw two graphs from these values. One graph should draw the recall precision graph per class. And the second graph should visualize an entire Precision-Recall across all classes.

So far I have not found anything about this topic in the Issues. But I would like to know the explanation of the structure of the two variables.

matejsuchanek commented 2 months ago

In case someone had the same problem and ended up here, this was answered in https://github.com/facebookresearch/maskrcnn-benchmark/issues/94#issuecomment-537690872.