DS-YOLOv8-Based Object Detection Method for Remote Sensing Images

被引:19
|
作者
Shen, Lingyun [1 ]
Lang, Baihe [2 ]
Song, Zhengxun [2 ,3 ]
机构
[1] Taiyuan Inst Technol, Dept Elect Engn, Taiyuan 030008, Peoples R China
[2] Changchun Univ Sci & Technol, Sch Elect & Informat Engn, Changchun 130022, Peoples R China
[3] Changchun Univ Sci & Technol, Overseas Expertise Intro Project Discipline Innova, Changchun 130022, Peoples R China
关键词
Object detection; deformable convolution; shuffle attention; self-calibration; wise-IoU; NETWORK;
D O I
10.1109/ACCESS.2023.3330844
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The improved YOLOv8 model (DCN_C2f+SC_SA+YOLOv8, hereinafter referred to as DS-YOLOv8) is proposed to address object detection challenges in complex remote sensing image tasks. It aims to overcome limitations such as the restricted receptive field caused by fixed convolutional kernels in the YOLO backbone network and the inadequate multi-scale feature learning capabilities resulting from the spatial and channel attention fusion mechanism's inability to adapt to the input data's feature distribution. The DS-YOLOv8 model introduces the Deformable Convolution C2f (DCN_C2f) module in the backbone network to enable adaptive adjustment of the network's receptive field. Additionally, a lightweight Self-Calibrating Shuffle Attention (SC_SA) module is designed for spatial and channel attention mechanisms. This design choice allows for adaptive encoding of contextual information, preventing the loss of feature details caused by convolution iterations and improving the representation capability of multi-scale, occluded, and small object features. Moreover, the DS-YOLOv8 model incorporates the dynamic non-monotonic focus mechanism of Wise-IoU and employs a position regression loss function to further enhance its performance. Experimental results demonstrate the excellent performance of the DS-YOLOv8 model on various public datasets, including RSOD, NWPU VHR-10, DIOR, and VEDAI. The average mAP@0.5 values achieved are 97.7%, 92.9%, 89.7%, and 78.9%, respectively. Similarly, the average mAP@0.5:0.95 values are observed to be 74.0%, 64.3%, 70.7%, and 51.1%. Importantly, the model maintains real-time inference capabilities. In comparison to the YOLOv8 series models, the DS-YOLOv8 model demonstrates significant performance improvements and outperforms other mainstream models in terms of detection accuracy.
引用
收藏
页码:125122 / 125137
页数:16
相关论文
共 50 条
  • [1] Object Detection Method Based on Improved YOLOv4 Network for Remote Sensing Images
    Xiao Zhenjiu
    Yang Yueying
    Kong Xiangxu
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (06)
  • [2] Efficient Object Detection Method Based on Improved YOLOv3 Network for Remote Sensing Images
    Wang, Jintao
    Xiao, Wen
    Ni, Tianwei
    [J]. 2020 3RD INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE AND BIG DATA (ICAIBD 2020), 2020, : 242 - 246
  • [3] Object Detection for Remote Sensing Based on the Enhanced YOLOv8 with WBiFPN
    Shen, Lingyun
    Lang, Baihe
    Song, Zhengxun
    [J]. IEEE Access, 2024, 12 : 158239 - 158257
  • [4] Object Detection Algorithm of Optical Remote Sensing Images Based on YOLOv3
    Wang Peng
    Xin Xuejing
    Wang Liqin
    Liu Rui
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (20)
  • [5] A Typical Remote Sensing object detection Method Based on YOLOv3
    Han, Y. C.
    Wang, J.
    Lu, L.
    [J]. 2019 4TH INTERNATIONAL CONFERENCE ON MECHANICAL, CONTROL AND COMPUTER ENGINEERING (ICMCCE 2019), 2019, : 520 - 523
  • [6] Small Object Detection Algorithm Based on Improved YOLOv8 for Remote Sensing
    Yi, Hao
    Liu, Bo
    Zhao, Bin
    Liu, Enhai
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2024, 17 : 1734 - 1747
  • [7] A NOVEL APPROACH TO OBJECT DETECTION IN REMOTE-SENSING IMAGES BASED ON YOLOv3
    Qin, Zhentao
    Tang, Yulin
    Jia, Yan
    Liu, Shi
    Yang, Ru
    Zhao, Xiangyu
    Zhang, Jin
    Mao, Xiaodong
    [J]. JOURNAL OF FLOW VISUALIZATION AND IMAGE PROCESSING, 2023, 30 (02) : 23 - 34
  • [8] YOLO-FNC: An Improved Method for Small Object Detection in Remote Sensing Images Based on YOLOv7
    Dang, Lanxue
    Liu, Gang
    Hou, Yan-e
    Han, Hongyu
    [J]. IAENG International Journal of Computer Science, 2024, 51 (09) : 1281 - 1290
  • [9] Improved YOLOv5-S object detection method for optical remote sensing images based on contextual transformer
    Zhou, Qikai
    Zhang, Wei
    Li, Ruizhi
    Wang, Jin
    Zhen, Shuhui
    Niu, Fu
    [J]. JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (04)
  • [10] Improved YOLOv5 Object Detection Algorithm for Remote Sensing Images
    Yang, Chen
    She, Lu
    Yang, Lu
    Feng, Zixian
    [J]. Computer Engineering and Applications, 2023, 59 (15) : 76 - 86