Apr, 2023

Align-DETR: 使用简单的 IoU 感知 BCE 损失来改进 DETR

TL;DR提出了一种改进的检测算法 Align-DETR,通过引入一个新的指标来解决 DETR 算法的 misalignment problem,该算法采用的 IA-BCE 损失函数可以有效训练定位精度与分类分数之间的关系,利用 prime sample weighting 机制抑制不重要样本的干扰,最终在 COCO 数据集上取得了 50.2% 的 AP 表现。