计算机科学与探索

作者：李青援, 邓赵红, 罗晓清, 顾鑫, 王士同

单位：1.江南大学 人工智能与计算机学院，江苏 无锡 214122；2.复旦大学 计算神经科学与类脑智能教育部重点实验室，上海 200433；3.张江实验室，上海 200120；4.江苏北方湖光光电有限公司，江苏 无锡 214035

关键词：目标检测,特征融合,注意力机制,深度学习

基金：国家自然科学基金面上项目(61772239);上海市市级重大科技专项(2018SHZDZX01)

为了进一步提升SSD算法的性能，解决SSD算法在进行多尺度预测时特征图信息不平衡和小目标识别难的问题，设计了即插即用的模块，充分融合不同尺度特征图包含的信息并建模特征图内的重要性关系，来增强特征图的表示能力。首先，设计了一种新颖的特征融合方法来解决跨尺度特征融合存在的信息差异问题。其次，根据池化金字塔的思想设计了一种深度特征提取模块来提取不同感受野的信息，从而提高模型对不同尺寸目标的检测能力。最后，为了进一步优化特征图，突出特征图对当前任务有效的信息，并建立全局像素点之间的长距离关系和各通道之间的重要性关系，提出了一种轻量级的注意力模块。通过上述机制，修改了SSD模型的架构，有效地提升了SSD算法的检测精度和鲁棒性。在PASCAL VOC数据集上设计了丰富的实验，验证了所提方法的有效性。在PASCAL VOC2007测试集上该方法比SSD算法提高了2.9个百分点的平均精确度（mAP），同时还保留了实时检测的能力。

来源：2022年第11期

《计算机科学与探索》期刊编辑部

查看计算机科学与探索杂志2022年第11期