• Laser & Optoelectronics Progress
  • Vol. 59, Issue 22, 2210004 (2022)
Hao Wang1、2、*, Zengshan Yin1、2, Guohua Liu1、2, Denghui Hu1, and Shuang Gao1、2
Author Affiliations
  • 1Innovation Academy for Microsatellite, Chinese Academy of Sciences, Shanghai 201203, China
  • 2University of Chinese Academy of Sciences, Beijing 100049, China
  • show less
    DOI: 10.3788/LOP202259.2210004 Cite this Article Set citation alerts
    Hao Wang, Zengshan Yin, Guohua Liu, Denghui Hu, Shuang Gao. Lightweight Object Detection Method for Optical Remote Sensing Image[J]. Laser & Optoelectronics Progress, 2022, 59(22): 2210004 Copy Citation Text show less

    Abstract

    In this paper, a lightweight optical remote sensing image target detection algorithm LW-YOLO is proposed based on the YOLOv5 detection model to solve the difficulty of deploying the deep learning target detection algorithm on the satellite due to the large volume of the model and too many parameters. First, a lightweight Ghost module is introduced to replace the ordinary convolution in the network to reduce the number of parameters and solve the computational overhead caused by feature information redundancy in the YOLOv5 network. Then, a space and channel Fusion Attention (FA) module is designed, and the bottleneck layer FABotleneck of the network is reconstructed to further reduce the parameters and improve the positioning ability of the algorithm for optical remote sensing image targets. Finally, a sparse parameter adaptive network pruning method is proposed to prune the network and further compress the model size. Experiments on the DOTA dataset show that compared with YOLOv5s, the LW-YOLO algorithm reduces 64.7% of parameters, 62.7% of model size, 3.7% of reasoning time, and only 6.4% of mean precision. The algorithm achieves the lightweight of the network model at the cost of small accuracy loss and provides a theoretical basis for on-orbit target detection in spaceborne optical images.
    Hao Wang, Zengshan Yin, Guohua Liu, Denghui Hu, Shuang Gao. Lightweight Object Detection Method for Optical Remote Sensing Image[J]. Laser & Optoelectronics Progress, 2022, 59(22): 2210004
    Download Citation