Opto-Electronic Engineering, Volume. 51, Issue 5, 240051(2024)

Improved YOLOv7 algorithm for target detection in complex environments from UAV perspective

Runmei Zhang1...2,3,4, Yufei Xiao1, Zhennan Jia1, Zhong Chen1,2, Zihua Chen1,2, Bin Yuan1,2,3,4, Weiwei Cao4 and Weiwei Song3,* |Show fewer author(s)
Author Affiliations
  • 1School of Mechanical and Electrical Engineering, Anhui Jianzhu University, Hefei, Anhui 230601, China
  • 2Key Laboratory of Intelligent Manufacturing of Construction Machinery, Hefei, Anhui 230601, China
  • 3Anhui Simulation Design and Modern Manufacturing Engineering Technology Research Center, Huangshan, Anhui 242700, China
  • 4Key Laboratory of Civil Aviation Flight Technology and Flight Safety, Guanghan, Sichuan 618300, China
  • show less

    To address the challenges faced by drones during UAV (unmanned aerial vehicle) photography in adverse conditions, such as low image recognition, obstruction by obstacles, and significant feature loss, a novel algorithm named SSG-YOLOv7 was proposed to enhance object detection from the perspective of drones in complex environments. Firstly, 12803 images were augmented from the VisDrone2019 dataset, and 1320 images were augmented from the RSOD dataset to simulate five different environments. Subsequently, anchor box sizes suitable for the datasets were clustered. The 3D non-local attention mechanism SimAM was integrated into the backbone network and feature extraction module to enhance the model's learning capabilities. Furthermore, the feature extraction module SPPCSPC was restructured to integrate information extracted from channels with different pool sizes and introduce the lightweight convolution module GhostConv, thereby improving the precision of dense multi-scale object detection without increasing the model's parameter count. Finally, Soft NMS was employed to optimize the confidence of anchor boxes, reducing false positives and missed detections. Experimental results demonstrate that SSG-YOLOv7 exhibits superior detection performance in complex environments, with performance metrics VisDrone_mAP@0.5 and RSOD_mAP@0.5 showing improvements of 10.45% and 2.67%, respectively, compared to YOLOv7.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Runmei Zhang, Yufei Xiao, Zhennan Jia, Zhong Chen, Zihua Chen, Bin Yuan, Weiwei Cao, Weiwei Song. Improved YOLOv7 algorithm for target detection in complex environments from UAV perspective[J]. Opto-Electronic Engineering, 2024, 51(5): 240051

    Download Citation

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

    Category: Article

    Received: Mar. 6, 2024

    Accepted: Apr. 24, 2024

    Published Online: Jul. 31, 2024

    The Author Email: Song Weiwei (宋娓娓)

    DOI:10.12086/oee.2024.240051

    Topics