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

some modifications added to rtdetrv2_torch.py #399

Closed mkhoshbin72 closed 1 month ago

mkhoshbin72 commented 1 month ago

There was an import error. The import from cvperception changed to src. Now the code should be run from the base directory of rtdetrv2_pytorch. The classes of detected boxes were printed wrong on the image. This bug is fixed. Also, I added the scores of the detected object to the saved image.