Infrared and Laser Engineering, Volume. 47, Issue 7, 703005(2018)

A lightweight small object detection algorithm based on improved SSD

Wu Tianshu1,2、*, Zhang Zhijia1, Liu Yunpeng2, Pei Wenhui1, and Chen Hongye1
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    In order to improve the small object detection ability of SSD object detection algorithm, the transposed convolution structure in SSD algorithm was proposed, the low resolution high semantic information feature map was integrated with high resolution low semantic information feature map using transposed convolution, which increased the ability of low level feature extraction and improved the average accuracy of SSD algorithm. At the same time for the problem that SSD algorithm model being large, running memory consumption high, without running on the embedded equipment ARM, a lightweight feature extraction minimum unit was proposed based on DenseNet, combining depthwise separable convolutions, pointwise group convolution and channel shuffle, running on the embedded equipment ARM cloud be realized. The comparative experiments on PASCAL VOC data set and KITTI autopilot data set show that the mean average is significantly improved by improved network structure, and the number of model parameters is effectively reduced.

    Tools

    Get Citation

    Copy Citation Text

    Wu Tianshu, Zhang Zhijia, Liu Yunpeng, Pei Wenhui, Chen Hongye. A lightweight small object detection algorithm based on improved SSD[J]. Infrared and Laser Engineering, 2018, 47(7): 703005

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: 特约专栏—“深度学习及其应用”

    Received: Feb. 10, 2018

    Accepted: Mar. 20, 2018

    Published Online: Aug. 30, 2018

    The Author Email: Tianshu Wu (747112077@qq.com)

    DOI:10.3788/irla201847.0703005

    Topics