NVIDIA-optimized C++ pycocotools extension

open-mmlab / cocoapi

COCO API - Dataset @ http://cocodataset.org/

Other

46 stars 35 forks source link

NVIDIA-optimized C++ pycocotools extension #11

Open ElectronicElephant opened 4 years ago

ElectronicElephant commented 4 years ago

Hello mmlab,

It's great to see COCO and LVIS API merged into one repo.

However, the original python api still suffers from slow evaluation speed (~ minutes). AFAIK, NVIDIA has brought a C++ extension (https://github.com/NVIDIA/cocoapi/blob/nvidia/master/PythonAPI/pycocotools/ext.cpp) , which can print the mAP & mAR table in less than one second, nearly in a flash. Would you consider introducing the extension into this repo?

hellock commented 4 years ago

Looks great. @xvjiarui Please take a look.

xvjiarui commented 4 years ago

Hi @ElectronicElephant Sorry for the late reply. We are planning to support it soon.