lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to use the onnx model to detect objects on a single image?

KyriakosChris opened this issue · comments

Hello, I used the I retrain the pytorch rtdetr_r18vd model for 72 epochs using the Visdrone Dataset. Then I converted the model into onnx and at the moment I want to use this model to detect objects on a single image. How do to it?

I figured it out, you can use the code that is commented out in the export.py file. It needs to edit it but it is almost complete