可视化分类和IOU 的散点图是如何做到的？

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.61k stars 303 forks source link

Open notfacezhi opened 11 months ago

notfacezhi commented 11 months ago

作者你好，我有一个问题不知道咋回事。如图所示一个query不是只有一个分类的置信度么？ IOU 置信度是从哪里来的？我不知道如何可视化上面这张图

lyuwenyu commented 11 months ago

嗯，这里是所有的query点 iou是和gt算出来的

notfacezhi commented 11 months ago

一个query 不是有多个gt么是和每个gt都算一遍，然后取均值？

lyuwenyu commented 11 months ago

选最大的代表这个query最终的预测结果。和推理的逻辑一样，