• Infrared and Laser Engineering
  • Vol. 47, Issue 7, 703005 (2018)
Wu Tianshu1、2、*, Zhang Zhijia1, Liu Yunpeng2, Pei Wenhui1, and Chen Hongye1
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less
    DOI: 10.3788/irla201847.0703005 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.
    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