计算机工程

作者：姜竣, 翟东海

单位：1. 西南交通大学 信息科学与技术学院, 成都 611756;2. 西藏大学 工学院, 拉萨 850000

关键词：卷积神经网络,SSD算法,特征融合,空洞卷积,目标检测

基金：国家自然科学基金（61961038）。

基于卷积神经网络目标检测算法的浅层特征图包含丰富的细节信息，但缺乏语义信息，而深层特征图则相反。为充分利用浅层和深层特征图特征，解决多尺度目标检测问题，提出一种新的单阶段目标检测算法（AFE-SSD）。以SSD算法为基础，分别对该算法中相邻的2个特征图进行特征融合，从而丰富浅层特征层的语义信息。通过对并行空洞卷积机制进行改进，构建多尺度特征提取模块，将融合后的特征图通入多尺度特征提取模块的方式丰富其多尺度信息，同时提升主干网络的特征提取能力。在PASCAL VOC2007测试集上的实验结果表明，AFE-SSD算法的mAP为79.8%，检测速度为58.8 frame/s，与SSD、DSSD算法相比，mAP分别提升了2.4和1.2个百分点，验证了所提特征融合方式及多尺度提取模块的有效性。

来源：2021年第7期

《计算机工程》期刊编辑部

查看计算机工程杂志2021年第7期