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

指标 #329

Open lei12879 opened 3 weeks ago

lei12879 commented 3 weeks ago

我想问一下rtdetr计算gflops和fps是如何计算的

lei12879 commented 3 weeks ago

pytorch版本的

lyuwenyu commented 3 weeks ago

pytorch版本的

可以使用torch.profiler.profile统计相关信息 https://pytorch.org/docs/stable/profiler.html