检测精度提升
通过交叉注意力的任务交互机制,显式提升分类与定位预测一致性,缓解单阶段检测器中的任务不对齐问题。
Ruilin, Yao · Rong, Yi · Qiangqiang, Huang · 熊盛武
IEEE Transactions on Circuits and Systems for Video Technology 2024
通过显式对齐分类与定位预测,在MS-COCO上单模型单尺度测试达到51.8 AP,超过近期单阶段检测器ATSS、VFNet、LD和TOOD。
设计双任务交互模块和空间特征聚合模块,生成任务交互嵌入并利用偏移量和权重聚合邻近特征点信息,从而对齐两个任务的预测结果。
代码已开源,可方便集成到现有检测流程或进行微调。