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] Multiscale Feature Fusion and Threshold-Based Attentional YOLO for Tailings Ponds Detection of Remote Sensing Images
    Zheng, Guizhou
    Zhao, Yingying
    Zhong, Junsheng
    Qiu, Zhonghang
    Chen, Zhixing
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2024, 21 : 1 - 5
  • [42] Remote sensing image ship detection based on modified YOLO algorithm
    Wang X.
    Jiang H.
    Lin K.
    Jiang, Hongxu (jianghx@buaa.edu.cn), 1600, Beijing University of Aeronautics and Astronautics (BUAA) (46): : 1184 - 1191
  • [43] FESSD: Feature Enhancement Single Shot MultiBox Detector Algorithm for Remote Sensing Image Target Detection
    Guo, Jianxin
    Wang, Zhen
    Zhang, Shanwen
    ELECTRONICS, 2023, 12 (04)
  • [44] Remote sensing image semantic segmentation method based on small target and edge feature enhancement
    Wang, Huaijun
    Qiao, Luqi
    Li, He
    Li, Xiujuan
    Li, Junhuai
    Cao, Ting
    Zhang, Chunyi
    JOURNAL OF APPLIED REMOTE SENSING, 2023, 17 (04)
  • [45] A YOLO Network Based on Depthwise Convolution Attention, Feature Fusion, and KL Divergence (DFK-YOLO): A Deep Learning Method for Infrared Small Target Detection Based on YOLOv7
    Ji, Peng
    Wu, Changhao
    Zhang, Xiangyue
    Liu, Hean
    He, Dongsheng
    ELECTRONICS, 2024, 13 (23):
  • [46] YOLO-FSD: An Improved Target Detection Algorithm on Remote-Sensing Images
    Zhao, Hu
    Chu, Kaibin
    Zhang, Ji
    Feng, Chengtao
    IEEE SENSORS JOURNAL, 2023, 23 (24) : 30751 - 30764
  • [47] FESSD:SSD target detection based on feature fusion and feature enhancement
    Qian, Huaming
    Wang, Huilin
    Feng, Shuai
    Yan, Shuya
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2023, 20 (01)
  • [48] FESSD:SSD target detection based on feature fusion and feature enhancement
    Huaming Qian
    Huilin Wang
    Shuai Feng
    Shuya Yan
    Journal of Real-Time Image Processing, 2023, 20
  • [49] Target Fusion Detection of LiDAR and Camera Based on the Improved YOLO Algorithm
    Han, Jian
    Liao, Yaping
    Zhang, Junyou
    Wang, Shufeng
    Li, Sixian
    MATHEMATICS, 2018, 6 (10)
  • [50] UAV remote sensing detection of tea leaf blight based on DDMA-YOLO
    Bao, Wenxia
    Zhu, Ziqiang
    Hu, Gensheng
    Zhou, Xingen
    Zhang, Dongyan
    Yang, Xianjun
    COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2023, 205