Drone detection is a significant research topic due to the potential security threats posed by the misuse of drones in both civilian and military domains. However, traditional drone detection methods are challenged by the drastic scale changes and complex ambiguity during drone flight, and it is difficult to detect small target drones quickly and efficiently. We propose an information-enhanced model based on improved YOLOv5 (TGC-YOLOv5) for fast and accurate detection of small target drones in complex environments. The main contributions of this paper are as follows: First, the Transformer encoder module is incorporated into YOLOv5 to augment attention toward the regions of interest. Second, the Global Attention Mechanism (GAM) is embraced to mitigate information diffusion among distinct layers and amplify the global cross-dimensional interaction features. Finally, the Coordinate Attention Mechanism (CA) is incorporated into the bottleneck part of C3, enhancing the extraction capability of local information for small targets. To enhance and verify the robustness and generalization of the model, a small target drone dataset (SUAV-DATA) is constructed in all-weather, multi-scenario, and complex environments. The experimental results show that based on the SUAV-DATA dataset, the AP value of TGC-YOLOv5 reaches 0.848, which is 2.5% higher than the original YOLOv5, and the Recall value of TGC-YOLOv5 reaches 0.823, which is a 3.8% improvement over the original YOLOv5. The robustness of our proposed model is also verified on the Real-World open-source image dataset, achieving the best accuracy in light, fog, stain, and saturation pollution images. The findings and methods of this paper have important significance and value for improving the efficiency and precision of drone detection.


    Access

    Download


    Export, share and cite



    Title :

    TGC-YOLOv5: An Enhanced YOLOv5 Drone Detection Model Based on Transformer, GAM & CA Attention Mechanism


    Contributors:
    Yuliang Zhao (author) / Zhongjie Ju (author) / Tianang Sun (author) / Fanghecong Dong (author) / Jian Li (author) / Ruige Yang (author) / Qiang Fu (author) / Chao Lian (author) / Peng Shan (author)


    Publication date :

    2023




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    Unknown




    STD-Yolov5: a ship-type detection model based on improved Yolov5

    Ning, Yue / Zhao, Lining / Zhang, Can et al. | Taylor & Francis Verlag | 2024


    Ship-YOLOv5s: improved YOLOv5 ship target detection based on attention mechanism

    Zhang, Peng / Sun, Ze / Dong, Junwei et al. | SPIE | 2025


    Road Traffic Vehicle Detection Method Using Lightweight YOLOv5 and Attention Mechanism

    Wang, Yunzhen / Ma, Hongbing / Li, Liangliang | IEEE | 2022


    TF-BiFPN Improves YOLOv5: Enhancing Small-Scale Multiclass Drone Detection in Dark

    Misbah, Maham / Orakazi, Farooq Alam / Tanveer, Laiba et al. | IEEE | 2025


    Runway Crack Detection Based on YOLOV5

    Li, Bu / Fu, Maoming / Li, Qin | IEEE | 2021