• • 下一篇
谢斌红,王瑞,张睿,张英俊
摘要: 针对现有小样本目标检测算法中类级原型生成精度不足、细节信息缺失导致目标区域特征表达能力受限的问题,提出了一种基于代理原型聚合(APA)的小样本目标检测算法(APA-FSOD),该算法通过代理注意力将支持特征蒸馏为细节丰富的原型,并基于相关性实现原型向量在查询特征图上的精准分配,显著强化了目标实例区域的特征表达能力。此外,还设计了小波卷积增强模块(WCEM)和自适应多关系融合模块(AMRF),分别用于优化算法的全局特征提取和高级特征融合。实验结果表明,在 PASCAL VOC 数据集的三种新类划分下,APA-FSOD 的 nAP50 相较于基线 VFA(Variational Feature Aggregation)最高提升了 1.1 个百分点。在 MS COCO 数据集的 30-shot 设置下,与元学习方法 SMPCCNet(Support-query Mutual Promotion and Classification Correction Network)相比,nAP 提升了 1 个百分点,显著提高了小样本目标检测的精度。
中图分类号: