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

关于pytorch版推理代码的问题 #282

Open super-song-sir opened 5 months ago

super-song-sir commented 5 months ago

作者大大，可以在后续的更新中给出pytorch版的图片可视化推理代码吗？新手小白运行完训练代码想看一下推理的图片结果，但是自己的水平写不出来推理代码。非常感谢

lyuwenyu commented 5 months ago

see https://github.com/lyuwenyu/RT-DETR/issues/95

super-song-sir commented 5 months ago

see #95

Sorry to bother you again, but do I have to transform the trained weights file through onnx before I can do inference? I didn't find any code for direct inference using pytorch. Could you please follow up with code that uses pytorch trained .pth files for direct visual test?

super-song-sir commented 5 months ago

见#95

Thank you,I find it at #136