【RT-DETR实战】064、NMS后处理优化与替代方案:我在RT-DETR里踩过的那些坑
今天调一个RT-DETR的部署问题,模型推理速度明明达标了,但在实际视频流里跟踪目标时总出现“闪跳”——同一个目标在相邻帧里忽左忽右。盯着输出看了半天,发现是相邻帧的检测框置信度相差0.01,NMS直接就把低分框干掉了,导致目标位置在帧间不连续。这个经典问题让我决定好好聊聊NMS后处理的那些事儿。传统NMS的硬伤标准NMS的逻辑简单粗暴:按置信度排序,最高分的框作为保留框,然后干掉所有和它IoU超过阈值的其他框。这个“干掉”是物理删除,直接置零。在RT-DETR这种密集预测的场景里,两个其实都是正确目标的框,可能因为角度、遮挡导致IoU略高,就被误杀了。# 典型NMS实现(问题版本)defnaive_nms(boxes,scores,iou_thresh=0.5)
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2636436.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!