• Laser Journal
  • Vol. 45, Issue 12, 49 (2024)
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
    DOI: 10.14016/j.cnki.jgzz.2024.12.049 Cite this Article
    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 Copy Citation Text show less

    Abstract

    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.
    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