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-0703005(7). DOI: 10.3788/IRLA201847.0703005
Citation: 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-0703005(7). DOI: 10.3788/IRLA201847.0703005

A lightweight small object detection algorithm based on improved SSD

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

Catalog

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return