IoU-Aware分类损失:把定位质量纳入分类评分,NMS误删率直降的秘籍
摘要:目标检测中,分类得分高≠定位准——这个“两张皮”问题让无数工程师在NMS阶段痛失好框。本文深入解析IoU-Aware分类损失的核心原理与工程实践,从VarifocalNet到Generalized Focal Loss再到Align-DETR,系统对比主流方案,附代码实现与部署建议,助你彻底告别NMS误删之痛。一、深夜加血的痛:那个被NMS“误杀”的好框做目标检测的兄弟,下面这个场景你一定不陌生——模型训练完,mAP看着还不错,一上真实场景就翻车。密集人群、车辆拥堵、货架堆叠,模型明明对两个目标都有响应,最后输出只剩一个框。问题出在哪?NMS(非极大值抑制)。标准NMS的逻辑很简单:按分类分数排序,保留最高分的框,干掉和它IoU超过阈值的其他框。代码就几行:defnms(boxes,scores,iou_threshold)
