[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
1747956LXR opened this issue 6 months ago · comments
我使用一张3090显卡复现pytorch版的rt-detr,复现的模型为rtdetr_r18vd_6x_coco.yml,设置的batchsize=16,其他参数和代码都没有改变,最后得到45.9的mAP,与论文报告的46.5的mAP还有差距,以下是训练最后几个epoch的截图,有劳作者帮忙看一下,谢谢!
这个版本我只跑了r50的 log放这里了#42
ps. 另外我重新设计了pytorch版本的rtdetrv2_r18的精度mAP 47.9
可以先star这个repo并持续关注一下后续的状态