任务对齐检测器

检测精度提升

通过交叉注意力的任务交互机制,显式提升分类与定位预测一致性,缓解单阶段检测器中的任务不对齐问题。

Ruilin, Yao · Rong, Yi · Qiangqiang, Huang · 熊盛武

IEEE Transactions on Circuits and Systems for Video Technology 2024

参数信息

检测精度 (AP)
51.8 AP

技术优势

精度更高

通过显式对齐分类与定位预测,在MS-COCO上单模型单尺度测试达到51.8 AP,超过近期单阶段检测器ATSS、VFNet、LD和TOOD。

任务对齐

设计双任务交互模块和空间特征聚合模块,生成任务交互嵌入并利用偏移量和权重聚合邻近特征点信息,从而对齐两个任务的预测结果。

易于使用

代码已开源,可方便集成到现有检测流程或进行微调。

应用场景