Laser Journal, Volume. 45, Issue 12, 49(2024)

Improved YOLOv5s Algorithm for rotation object detection in remote sensing images

LIU Bingbing1, HU Yaoguo1, YAN Peng2, and ZHANG Qinlin1、*
Author Affiliations
  • 1College of Physical Science and Technology, Central China Normal University, Wuhan 430079, China
  • 2Xi'an Institute of Optics and Precision Mechanics of CAS, Xi'an 710068, China
  • show less

    Remote sensing images object detection is used in mineral exploration, transportation, national defense and military, emergency rescue and disaster relief and other fields. However, the common rotation object detection algorithm model applicable to remote sensing images is too large, difficult to deploy and can not meet the requirements of real-time detection, ignoring the balance between accuracy and speed. To solve the above problems, a rotation object detection algorithm YOLOV5s-R based on YOLOv5s is proposed. Firstly, angle parameters were added on the basis of YOLOv5s, then the horizontal bounding box was modified into oriented bounding box to adapt to the angular diversity of remote sensing image objects. Secondly, the Circular Smooth Label was introduced to avoid the angle mutation problem caused by the periodicity of angle regression. Then, the Efficient Channel Attention module was introduced to improve the ability of the model to extract important features. Finally, the Adaptively Spatial Feature Fusion module was introduced to solve the inconsistencies between different feature scales inside the feature pyramid. On the dataset DOTA, the experimental results show that the mAP50 reaches 75.6%, the mAP50∶95 reaches 46.7%, and the FPS reaches 81.9. Compared with the base model, mAP50 and mAP50∶95 increased by 1% and 3.1% respectively, and FPS increased by 85.9%. Therefore, YOLOv5s-R achieves more accurate and high-speed remote sensing images detection, achieving a good balance between accuracy and speed.

    Tools

    Get Citation

    Copy Citation Text

    LIU Bingbing, HU Yaoguo, YAN Peng, ZHANG Qinlin. Improved YOLOv5s Algorithm for rotation object detection in remote sensing images[J]. Laser Journal, 2024, 45(12): 49

    Download Citation

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

    Category:

    Received: Mar. 29, 2024

    Accepted: Mar. 10, 2025

    Published Online: Mar. 10, 2025

    The Author Email: Qinlin ZHANG (zhangqinglin@mail.ccnu.edu.cn)

    DOI:10.14016/j.cnki.jgzz.2024.12.049

    Topics