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
1.64k stars 178 forks source link

FPS相关问题 #299

Open txw-github opened 1 month ago

txw-github commented 1 month ago

我只修改了数据集,为什么我在RTX3060上使用pytorch版本configs/rtdetr/rtdetr_r18vd_6x_coco.yml,训练数据集6000张需要4-5min,测试集2000张需要45s。而yolov8只要1.3min和17s,请求大佬指导下,比较急需

lyuwenyu commented 1 month ago

所以具体有什么问题?

txw-github commented 1 month ago

所以具体有什么问题?

就是论文中不是说速度比yolov8快嘛,但是我这操作结果慢了很多

lyuwenyu commented 1 month ago

模型的端到端速度的定义请参考

模型测速代码请参考

txw-github commented 1 month ago

模型的端到端速度的定义请参考

  • 原文的 3.2. End-to-end Speed Benchmark

模型测速代码请参考

懂了,我去试试,太感谢您了