Open MiXaiLL76 opened 3 weeks ago
@BIGWangYuDong Hello! I'm waiting for you Yudong Wang
Good library to speed up calculations
LGTM, kindly ping @hhaAndroid have a look
Hello @hhaAndroid, I see that you are active on the site, please take a look at this PR
@RangiLyu Maybe you’re here and the project hasn’t been abandoned yet?
Motivation
Faster COCO Eval
https://github.com/MiXaiLL76/faster_coco_eval
This package wraps a facebook C++ implementation of COCO-eval operations found in the pycocotools package. This implementation greatly speeds up the evaluation time for coco's AP metrics, especially when dealing with a high number of instances in an image.
Modification
I added a variable (use_faster_coco_eval) to the CocoMetric class that allows you to select an additional type of validation, namely COCOeval_faster
Use cases (Optional)
For our use case with a test dataset of 5000 images from the coco val dataset. Testing was carried out using the mmdetection framework and the eval_metric.py script. The indicators are presented below.
Visualization of testing colab_example.ipynb available in directory examples/comparison colab_example.ipynb in google collab Tested with rtmdet model bbox + segm
Summary for 5000 imgs
Checklist