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.53k stars 292 forks source link

IoU of model inferencing #283

Open vishalvijay99 opened 6 months ago

vishalvijay99 commented 6 months ago

While calling the model for prediction on image what IoU threshold is used?I don't see any number or counter to set threshold for IoU. Where to find this?

lyuwenyu commented 6 months ago

RTDETR don't have nms (so no IoU threshold) in postprocessor.

DoubleDa-TWJ commented 2 months ago

RTDETR 在后处理器中没有 nms（所以没有IoU threshold）。

那如果我想在测试的时候设置IoU阈值为0.5的话应该怎么办呢？