Bulletin of Surveying and Mapping ›› 2022, Vol. 0 ›› Issue (10): 37-43,55.doi: 10.13474/j.cnki.11-2246.2022.0291

Previous Articles     Next Articles

A lightweight dense connection network for object detection of remote sensing images

LIU Ji1,2,3, YANG Jun1,2,3   

  1. 1. Faculty of Geomatics, Lanzhou Jiaotong University, Lanzhou 730070, China;
    2. National-Local Joint Engineering Research Center of Technologies and Applications for National Geographic State Monitoring, Lanzhou 730070, China;
    3. Gansu Provincial Engineering Laboratory for National Geographic State Monitoring, Lanzhou 730070, China
  • Received:2021-11-09 Published:2022-11-02

Abstract: Two issues arising out of the small objects, detected within UAV remote sensing images, sometimes happening to be multiple and dense increases in both the misdetection and miss out rates. A lightweight target detection algorithm is proposed based on YOLO v4 with a dense connection network module to realize high precision identification of the small targets. Firstly, feature reuse and feature extraction enhancement are facilitated by adjusting the convolutional layers of CSP Darknet53 (backbone network of YOLO v4) to work in two modes of actions:dense connection and sparse connection; the gradient disappearance problem is also alleviated. Secondly, the proposed module tailors the models to fit into a reduced number of network layers; then the resulting module is redefined as the new densely connected network module. Experiments are carried out on the NWPU-VHR-10 and Vihicle-850 UAV image datasets, a courtesy of our research group. Our model has the upper hand in terms of the number of the accurately detected small targets within the remote sensing images; our implementation is also cost-effective in terms of the network model convergence time and memory consumption. A significant improvement in the speed of detection is gained.

Key words: object detection, unmanned aerial vehicle remote sensing images, YOLO v4, lightweight, dense connection

CLC Number: