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.18k stars 237 forks source link

There are multiple bounding boxes on a single object. #379

Open Ian-Work-AI opened 1 month ago

Ian-Work-AI commented 1 month ago

The model works well in most cases, but sometimes it detects multiple bounding boxes for the same single object. The DETR paper says that the model will generate an independent bounding box for each individual target, so they didn't need NMS. Shouldn't the output keep only the highest scoring bounding box?

Thanks for your help!!

PaulineTreyvaud commented 1 month ago

I do have the same issue too. I obtain several perfectly overlapping boxes with different classes and even boxes with high IoU with the same label. Have you ever noticed that @lyuwenyu ?

lyuwenyu commented 1 month ago

I think you should choose score threshold carefully.

David-dotcom666 commented 2 weeks ago

@lyuwenyu 应该不是分数阈值设置合理性的问题吧,会出现分数非常相近的几乎重叠的情况 1 2 3