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 259 forks source link

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

Closed KyriakosChris closed 8 months ago

KyriakosChris commented 8 months ago

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?

KyriakosChris commented 8 months ago

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