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.31k stars 258 forks source link

onnx和pytorch版本的推理速度 #356

Open Eddie-cc opened 2 months ago

Eddie-cc commented 2 months ago

Thanks your Great Work

是否可以提供onnx和pytorch版本模型对应的推理时间或fps，最终trt版本的推理速度和论文没法对齐，想逐步排查问题所在

lyuwenyu commented 2 months ago

论文测速参考paddle版本文档 https://github.com/lyuwenyu/RT-DETR/tree/main/rtdetr_paddle

image
Eddie-cc commented 2 months ago

感谢回复，我已参考文档完成转换，不过在测试速度上存在疑惑

是否可以向我提供您测试出的推理速度（pytorch版本和onnx版本）以及相应的gpu型号

lyuwenyu commented 2 months ago

文档和论文里有详细说明

image