pp. 3059-3073
S&M3718 Research Paper of Special Issue https://doi.org/10.18494/SAM5136 Published: July 31, 2024 Improved YOLOv5 Algorithm for Oriented Object Detection of Aerial Image [PDF] Gang Yang, Miao Wang, Quan Zhou, Jiangchuan Li, Siyue Zhou, and Yutong Lu (Received May 7, 2024; Accepted July 4, 2024) Keywords: object detection, remote sensing image, deep learning, YOLO
With the development of computer vision and remote sensor devices, object detection in aerial images has drawn considerable attention because of its ability to provide a wide field of view and a large amount of information. Despite this, object detection in aerial images is a challenging task owing to densely packed objects, oriented diversity, and complex background. In this study, we optimized three aspects of the YOLOv5 algorithm to detect arbitrary oriented objects in remote sensing images, including head structure, features from the backbone, and angle prediction. To improve the head structure, we decoupled it into four submodules, which are used for object localization, foreground, category, and oriented angle classification. To increase the accuracy of the features from the backbone, we designed a block dimensional attention module, which is developed by splitting the image into smaller patches based on a dimensional attention module. Compared with the original YOLOv5 algorithm, our approach has a better performance for oriented object detection—the mAP on DOTA-v1.5 is increased by 1.25%. It was tested to be effective on DOTA-v1.0, HRSC2016, and DIOR-R datasets as well.
Corresponding author: Miao Wang![]() ![]() This work is licensed under a Creative Commons Attribution 4.0 International License. Cite this article Gang Yang, Miao Wang, Quan Zhou, Jiangchuan Li, Siyue Zhou, and Yutong Lu, Improved YOLOv5 Algorithm for Oriented Object Detection of Aerial Image, Sens. Mater., Vol. 36, No. 7, 2024, p. 3059-3073. |