计算机科学与探索 ›› 2019, Vol. 13 ›› Issue (6): 1049-1061.DOI: 10.3778/j.issn.1673-9418.1810005

• 图形图像 • 上一篇    下一篇

多尺度卷积特征融合的SSD目标检测算法

陈幻杰1,王琦琦1+,杨国威1,韩佳林1,尹成娟1,陈  隽2,王以忠1   

  1. 1.天津科技大学 电子信息与自动化学院,天津 300222
    2.麦克马斯特大学 电子与计算机工程系,加拿大 汉密尔顿 L8E、L8W
  • 出版日期:2019-06-01 发布日期:2019-06-14

SSD Object Detection Algorithm with Multi-Scale Convolution Feature Fusion

CHEN Huanjie1, WANG Qiqi1+, YANG Guowei1, HAN Jialin1, YIN Chengjuan1, CHEN Jun2, WANG Yizhong1   

  1. 1. College of Electronic Information and Automation, Tianjin University of Science and Technology, Tianjin 300222, China
    2. Department of Electrical and Computer Engineering, McMaster University, Hamilton L8E, L8W, Canada
  • Online:2019-06-01 Published:2019-06-14

摘要: 提出了一种改进的多尺度卷积特征目标检测方法,用以提高SSD(single shot multibox detector)模型对中目标和小目标的检测精确度。该方法先对SSD模型低层特征层采用区域放大提取的方法以提高对小目标的检测能力,再对高层特征层进行特征提取以改善中目标的检测效果。最后,利用SSD模型中原有的多尺度卷积检测方法,将改进的多层特征检测结果进行融合,并通过参数再训练以获得最终改进的SSD模型。实验结果表明,该方法在MS COCO数据集上对中目标和小目标的测试精确度分别为75.1%和40.5%,相比于原有SSD模型分别提升16.3%和23.1%。

关键词: 单次多框目标检测器(SSD)模型, 多尺度特征融合, 目标检测, 深度学习

Abstract: In this paper, it is proposed that the accuracy of small and medium objects detection of SSD (single shot multibox detector) can be improved by incorporating a modified multi-scale convolution feature fusion method. In order to enhance small objects detection, the region magnification extraction on the low-layer feature maps is adopted. Then, features are extracted from the high-layer feature maps to make the detection of medium objects better. These features are finally fused by the multi-scale convolution detection method as in the original SSD architecture. Moreover, the parameters of the present model are obtained via the parameter retraining. On the MS COCO test set, the present model shows that the mAP (mean average precision) of medium and small objects decetion is 75.1% and 40.5% respectively, which is nearly 16.3% and 23.1% higher than the original SSD.

Key words: single shot multibox detector (SSD), multi-scale feature fusion, object detection, deep-learning