dddzg / up-detr

[TPAMI 2022 & CVPR2021 Oral] UP-DETR: Unsupervised Pre-training for Object Detection with Transformers
Apache License 2.0
476 stars 71 forks source link

How extract precision, recall and f1-score metrics #17

Closed jackonealll closed 2 years ago

jackonealll commented 2 years ago

Hello thank you for sharing the code.

I would like to know how to extract precision, recall and f1-score metrics. I already have the AP and AR metrics.

I am trying to use the following code but it gives me a numpy matrix:

precision = coco_eval.eval['precision']
recall = coco_eval.eval['recall']

Can you help me?

dddzg commented 2 years ago

I am sorry that this issue is not related to our work. I guess you can find more details in pycocotools