search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.33k stars 265 forks source link

Can pytorch RT-DETR calculate class-wise map? #53

Open Zhouziyuya opened 1 year ago

Zhouziyuya commented 1 year ago

Does pytorch version can output map of each class like paddle version when evaluating the model?

lyuwenyu commented 1 year ago

Not yet ~

Zhouziyuya commented 1 year ago

Will this be updated recently? Thanks for your masterpiece!

lyuwenyu commented 1 year ago

I will add it into TODO-LIST, but I can only do this during off hours, so there is no strict ddl.

You can using some off-shelf tool, like this https://github.com/laclouis5/globox

Zhouziyuya commented 1 year ago

Thanks!! I have changed some codes in pycocotools.cocoeval and make it.

mengdan-tan commented 7 months ago

Thanks!! I have changed some codes in pycocotools.cocoeval and make it.

Could you share how to change codes in pycocotools.cocoeval?Thanks!

xyb1314 commented 2 months ago

Thanks!! I have changed some codes in pycocotools.cocoeval and make it.

waiting for the class-wise map code 😘 thanks!

xyb1314 commented 2 months ago

Thanks!! I have changed some codes in pycocotools.cocoeval and make it.

Could you share how to change codes in pycocotools.cocoeval?Thanks!

Have you finished this implement? waiting for your shares!