计算机工程 ›› 2020, Vol. 46 ›› Issue (1): 247-254.doi: 10.19678/j.issn.1000-3428.0053233

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

基于多尺度融合SSD的小目标检测算法

赵亚男, 吴黎明, 陈琦   

  1. 广东工业大学 机电工程学院, 广州 510006
  • 收稿日期:2018-11-23 修回日期:2019-01-23 出版日期:2020-01-15 发布日期:2019-03-05
  • 作者简介:赵亚男(1992-),女,硕士研究生,主研方向为图像处理、人工智能;吴黎明,教授;陈琦,硕士研究生。
  • 基金项目:
    国家自然科学基金(61705045)。

Small Object Detection Algorithm Based on Multi-Scale Fusion SSD

ZHAO Ya'nan, WU Liming, CHEN Qi   

  1. School of Electromechanical Engineering, Guangdong University of Technology, Guangzhou 510006, China
  • Received:2018-11-23 Revised:2019-01-23 Online:2020-01-15 Published:2019-03-05

摘要: 针对一阶段目标检测算法在识别小目标时无法兼顾精度与实时性的问题,提出一种基于多尺度融合单点多盒探测器(SSD)的小目标检测算法。以SSD和DSSD算法的网络结构为基础,设计融合模块以实现Top-Down结构的功能,形成高层网络与低层网络之间的跳跃连接,结合SSD-VGG16扩展卷积特征图以提取多尺度特征,并对不同卷积层、尺度及特征的多元信息进行分类预测与位置回归。在织物瑕疵数据库上的实验结果表明,与SSD、DSSD等算法相比,该算法的检测性能较好,其检测精度达到78.2%,检测速度为51 frame/s,能在保证检测精度的同时提高检测速度。

关键词: 单点多盒探测器, 多尺度融合, 目标检测, 小目标, VGG16网络结构

Abstract: One-stage detection algorithms cannot balance precision and real-time performance in small object detection.To address the problem,this paper proposes a small object detection algorithm based on multi-scale fusion Single Shot multi-box Detector(SSD).The algorithm designs a fusion module based on the network structure of SSD and Deconvolutional Single Short Detector(DSSD) algorithms to implement functions of Top-Down structure,and thus enables skip connections between high-level network and low-level network.Then SSD-VGG16 is used to extend convolution feature map to extract multi-scale features,and multivariate data of different convolutional layers,scales and features is classified for prediction and position regression.Experimental results on a fabric defect database show that the detection precision of the proposed algorithm reaches 78.2% and detection speed reaches 51 frame/s,which outperforms SSD,DSSD and other algorithms.The results prove that the proposed algorithm can improve the detection speed with a high precision ensured.

Key words: Single Shot multi-box Detector(SSD), multi-scale fusion, object detection, small object, VGG16 network structure

中图分类号: