ICGM-YOLO: UAV remote sensing small target detection algorithm based on feature fusion enhancement module and ghost convolution

被引:0
|
作者
Zhang, Hongbin [1 ,2 ]
Yang, Jingmin [1 ,2 ,3 ]
Zhang, Wenjie [1 ,2 ]
Ren, Jinghui [1 ,2 ]
机构
[1] Minnan Normal Univ, Sch Comp Sci, Zhangzhou 363000, Fujian, Peoples R China
[2] Fujian Prov Univ, Key Lab Data Sci & Intelligence Applicat, Zhangzhou 363000, Fujian, Peoples R China
[3] Natl Taipei Univ Technol, Dept Elect Engn, Taipei, Taiwan
关键词
small target detection; YOLOv8; deep learning; remote sensing images; OBJECT DETECTION;
D O I
10.1088/1402-4896/ad92b0
中图分类号
O4 [物理学];
学科分类号
0702 ;
摘要
Remote sensing images captured at high altitudes are affected by background noise and low resolution, leading to suboptimal performance of commonly used small object detection algorithms. This paper introduces ICGM-YOLO, an enhanced approach designed to improve the precision of detecting small objects within remote sensing images. The methodology refines YOLOv8 for optimal performance in this context. First, within the feature extraction network, iterative attention feature fusion combines feature maps and integrates features from different layers. Second, the convolutional layer and C2f module are replaced with the GhostConv module and C3Ghost to extract features. Finally, this paper replaces CIoU with the MFIoU loss function, which combines the new bounding box MPDIoU based on minimum point distance with Focaler-IoU. This replacement accelerates model convergence and enhances detection recall. Experimental results from the Visdrone2019 dataset indicate that ICGM-YOLO outperforms the original YOLOv8 by achieving an 8.1% improvement in mAP0.5 detection rate and a 5.3% increase in mAP0.5:0.95 detection rate. Additionally, ICGM-YOLO reduces the parameter count by 48.2% and decreases computational complexity by 45.2%.
引用
收藏
页数:17
相关论文
共 50 条
  • [41] Research on target tracking algorithm based on multi -layer convolution feature fusion
    Xiao, Kehao
    Hao, Zhou
    2022 INTERNATIONAL CONFERENCE ON INDUSTRIAL AUTOMATION, ROBOTICS AND CONTROL ENGINEERING, IARCE, 2022, : 1 - 8
  • [42] YOLO-DCTI: Small Object Detection in Remote Sensing Base on Contextual Transformer Enhancement
    Min, Lingtong
    Fan, Ziman
    Lv, Qinyi
    Reda, Mohamed
    Shen, Linghao
    Wang, Binglu
    REMOTE SENSING, 2023, 15 (16)
  • [43] Small target detection with remote sensing images based on an improved YOLOv5 algorithm (vol 16, 1074862, 2023)
    Pei, Wenjing
    Shi, Zhanhao
    Gong, Kai
    FRONTIERS IN NEUROROBOTICS, 2023, 17
  • [44] Remote sensing image target detection based on a multi-scale deep feature fusion network
    Fan X.
    Yan W.
    Shi P.
    Zhang X.
    National Remote Sensing Bulletin, 2022, 26 (11): : 2292 - 2303
  • [45] YOLO-SD: Small Ship Detection in SAR Images by Multi-Scale Convolution and Feature Transformer Module
    Wang, Simin
    Gao, Song
    Zhou, Lun
    Liu, Ruochen
    Zhang, Hengsheng
    Liu, Jiaming
    Jia, Yong
    Qian, Jiang
    REMOTE SENSING, 2022, 14 (20)
  • [46] SMFF-YOLO: A Scale-Adaptive YOLO Algorithm with Multi-Level Feature Fusion for Object Detection in UAV Scenes
    Wang, Yuming
    Zou, Hua
    Yin, Ming
    Zhang, Xining
    REMOTE SENSING, 2023, 15 (18)
  • [47] LGFF-YOLO: small object detection method of UAV images based on efficient local-global feature fusion
    Peng, Hongxing
    Xie, Haopei
    Liu, Huanai
    Guan, Xianlu
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (05)
  • [48] DSAA-YOLO: UAV remote sensing small target recognition algorithm for YOLOV7 based on dense residual super-resolution and anchor frame adaptive regression strategy
    Hui, Yanming
    Wang, Jue
    Li, Bo
    JOURNAL OF KING SAUD UNIVERSITY-COMPUTER AND INFORMATION SCIENCES, 2024, 36 (01)
  • [49] A Feature Enhancement and Augmentation-Based Infrared Small Target Detection Network
    Chen, Siyang
    Wang, Han
    Shen, Zhihua
    Zhang, Guoyi
    Ning, Chenghao
    Zhang, Xiaohu
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21
  • [50] Lightweight Context Awareness and Feature Enhancement for Anchor-Free Remote- Sensing Target Detection
    Fan, Fei
    Zhang, Ming
    Yu, Dahua
    Li, Jianjun
    Zhou, Shichuang
    Liu, Yang
    IEEE SENSORS JOURNAL, 2024, 24 (07) : 10714 - 10726