Remote-Sensing Image Object Detection Based on Improved YOLOv8 Algorithm

被引:1
|
作者
Zhang Xiuzai [1 ,2 ]
Shen Tao [1 ]
Xu Dai [1 ]
机构
[1] Nanjing Univ Informat Sci Technol, Sch Elect & Informat Engn, Nanjing 210044, Jiangsu, Peoples R China
[2] Nanjing Univ Informat Sci Technol, Jiangsu Prov Atmospher Environm & Equipment Techn, Nanjing 210044, Jiangsu, Peoples R China
关键词
target detection; YOLOv8; WIoU; global attention mechanism; deformable convolution;
D O I
10.3788/LOP231803
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
A target detection algorithm based on improved YOLOv8 is proposed to address the issues of high-missed and false-detection rates, inaccurate target positioning, and inability to accurately identify target categories in remote-sensing image target detection algorithms. To improve the flexibility of the loss function of the model in gradient allocation and adapt to various object shapes and sizes, a boundary box regression loss function is designed, which combines a nonmonotonic focusing mechanism with geometric factors of the boundary box. To expand the receptive field of the model and weaken the influence of the remote-sensing image background on the detection target, a residual global attention mechanism is designed by combining global attention mechanism and residual blocks. To adapt the model to the deformation and irregular arrangement of target objects in remote-sensing images, the C2f module in the YOLOv8 model is improved by incorporating deformable convolution and deformable region-of-interest pooling layers. Experimental results show that on DOTA and RSOD datasets, mean average precision (mAP@0.5) of the improved YOLOv8 algorithm reaches 72.1% and 94.6%, which are better than other mainstream algorithms. It improves the accuracy of remote sensing image target detection and provides a new means for remote sensing image target detection.
引用
收藏
页数:11
相关论文
共 22 条
  • [1] Deformable Convolutional Networks
    Dai, Jifeng
    Qi, Haozhi
    Xiong, Yuwen
    Li, Yi
    Zhang, Guodong
    Hu, Han
    Wei, Yichen
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 764 - 773
  • [2] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [3] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [4] Hoffmann Nico, 2021, arXiv, DOI DOI 10.48550/ARXIV.2112.05561
  • [5] Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
  • [6] Improved YOLOv5 for Remote Sensing Image Detection
    Liu, Tao
    Ding, Xueyan
    Zhang, Bingbing
    Zhang, Jianxin
    [J]. Computer Engineering and Applications, 2023, 59 (10) : 253 - 261
  • [7] SSD: Single Shot MultiBox Detector
    Liu, Wei
    Anguelov, Dragomir
    Erhan, Dumitru
    Szegedy, Christian
    Reed, Scott
    Fu, Cheng-Yang
    Berg, Alexander C.
    [J]. COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 : 21 - 37
  • [8] DC-YOLOv8: Small-Size Object Detection Algorithm Based on Camera Sensor
    Lou, Haitong
    Duan, Xuehu
    Guo, Junmei
    Liu, Haiying
    Gu, Jason
    Bi, Lingyun
    Chen, Haonan
    [J]. ELECTRONICS, 2023, 12 (10)
  • [9] Efficient non-maximum suppression
    Neubeck, Alexander
    Van Gool, Luc
    [J]. 18TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION, VOL 3, PROCEEDINGS, 2006, : 850 - +
  • [10] Redmon J., 2018, arXiv, DOI 10.48550/arXiv.1804.02767