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.21k stars 242 forks source link

Pytorch RTDETR #42

Open lyuwenyu opened 1 year ago

lyuwenyu commented 1 year ago

code

https://github.com/lyuwenyu/RT-DETR/tree/main/rtdetr_pytorch

weight

rtdetr_r18vd_5x_coco_objects365_from_paddle.pth rtdetr_r18vd_1x_objects365_from_paddle.pth

rtdetr_r50vd_6x_coco_from_paddle.pth rtdetr_r50vd_2x_coco_objects365_from_paddle.pth rtdetr_r50vd_1x_objects365_from_paddle.pth

rtdetr_r101vd_6x_coco_from_paddle.pth rtdetr_r101vd_2x_coco_objects365_from_paddle.pth rtdetr_r101vd_1x_objects365_from_paddle.pth

logs

log_rtdetr_r50vd_coco_pytorch.txt

Zhao-Yian commented 1 year ago

Thanks for your tireless efforts on our behalf!

flyingbird9 commented 1 year ago

May I ask if rt detr has used dynamic anchor box updates for dab-detr?

rafaelpadilla commented 11 months ago

Thanks for RT-DETR :)

It would be so valuable if the links for the weights were in the README.md.

ramonhollands commented 4 months ago

Thanks for this amazing repo! Are there also configs/weights available for the scaled R18 used in the paper? Thanks in advance! Regards, Ramon

ramonhollands commented 3 months ago

Is the only change indeed changing the depth_mult to 0.33 and the eval_idx to 2 to reproduce the scaled R18 in the paper?