Laser & Optoelectronics Progress, Volume. 59, Issue 18, 1828003(2022)

Object Detection of Individual Mangrove Based on Improved YOLOv5

Yongkang Ma1,2, Hua Liu1,2、*, Chengxing Ling1,2, Feng Zhao1,2, Yi Jiang1,2, and Yutong Zhang1,2
Author Affiliations
  • 1Research Institute of Forest Resource Information Techniques, Chinese Academy of Forestry, Beijing 100091, China
  • 2Key Laboratory of Forestry Remote Sensing and Information System, National Forestry and Grassland Administration, Beijing 100091, China
  • show less

    In this study, an individual mangrove object detection model called YOLOv5-ECA based on deep learning is proposed to automatically identify and locate individual mangroves with high accuracy aiming at the challenges of small and dense individual mangroves in drone images, resulting in low automation and efficiency for detecting them. First, the open-source software LabelImg is used to mark the target tree on the selected drone image, which is applied to construct the individual mangrove dataset. Then, the YOLOv5 is used as the basic object detection model to maximize and enhance the target tree, and achieving this is based on the characteristics of dense distribution and small size of objects. The efficient channel attention (ECA) mechanism enhances the CSPDarknet53 backbone network to avoid dimensionality reduction while enhancing feature expression capabilities. Furthermore, the enhanced SoftPool pooling operation is introduced into the SPP module to retain more detailed feature information. Finally, the ACON adaptive activation function determines whether the neuron is activated. The results demonstrate that the constructed dataset is used to train the network before and after improvement, and the accuracy, recall, and mean average precision (mAP)@0.5 parameters are compared. The results of different models are slightly different, but they all tend to converge. The proposed YOLOv5-ECA's average detection accuracy is 3.2 percentage points higher than YOLOv5 and 5.19 percentage points higher than YOLOv4, and its training loss is also lower. The deep learning-based YOLOv5-ECA model can quickly, accurately, and automatically detect individual mangroves and significantly enhance the ability to identify and locate them.

    Tools

    Get Citation

    Copy Citation Text

    Yongkang Ma, Hua Liu, Chengxing Ling, Feng Zhao, Yi Jiang, Yutong Zhang. Object Detection of Individual Mangrove Based on Improved YOLOv5[J]. Laser & Optoelectronics Progress, 2022, 59(18): 1828003

    Download Citation

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

    Category: Remote Sensing and Sensors

    Received: Jul. 18, 2021

    Accepted: Aug. 10, 2021

    Published Online: Aug. 29, 2022

    The Author Email: Liu Hua (liuhua@ifrit.ac.cn)

    DOI:10.3788/LOP202259.1828003

    Topics