Author Login Editor-in-Chief Peer Review Editor Work Office Work

Computer Engineering ›› 2020, Vol. 46 ›› Issue (1): 247-254. doi: 10.19678/j.issn.1000-3428.0053233

Previous Articles     Next Articles

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的小目标检测算法

赵亚男, 吴黎明, 陈琦   

  1. 广东工业大学 机电工程学院, 广州 510006
  • 作者简介:赵亚男(1992-),女,硕士研究生,主研方向为图像处理、人工智能;吴黎明,教授;陈琦,硕士研究生。
  • 基金资助:
    国家自然科学基金(61705045)。

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

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

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

CLC Number: