To solve the issues of low accuracy in detecting small targets in drone aerial images and the excessive computational complexity of the detection model, this paper proposes a drone small target detection method based on an enhanced version of YOLOv5. Firstly, in an endeavor to ameliorate the issue of undetected instances of small-scale targets, this study incorporates the Convolutional Block Attention Module (CBAM) into the feature extraction framework of YOLOv5s. This augmentation enhances the capacity for extracting features from diminutive targets, leading to increased accuracy in small target detection. Secondly, in response to the issue of high computational complexity in the model, this paper has improved the CBL module by replacing regular convolutions with Coord convolutions. This modification helps reduce the model's parameters and computational complexity, resulting in a more efficient and streamlined detection process. Lastly, this paper utilizes the EIOU Loss function to further improve the accuracy and convergence rate of target detection. The amalgamation of these improvements markedly enhances the model's performance. This paper conducted tests using the DOTA dataset, and the experimental results demonstrate the effectiveness of the CCE-YOLOv5s model. It outperforms the original model with a mAp (mean Average Precision) of 80%, represents a noteworthy 1% enhancement over the original algorithm. Moreover, the training time is reduced to 0.66 hours, a remarkable 43% reduction in comparison to the original algorithm. Overall, this improved model effectively improves detection accuracy while ensuring real-time performance, and significantly improves the training speed of the model.


    Zugriff

    Zugriff prüfen

    Verfügbarkeit in meiner Bibliothek prüfen

    Bestellung bei Subito €


    Exportieren, teilen und zitieren



    Titel :

    CCE-YOLOv5s: An Improved YOLOv5 Model for UAV Small Target Detection


    Beteiligte:
    Zhu, Deren (Autor:in) / Dai, Linfeng (Autor:in) / Du, Peirong (Autor:in)


    Erscheinungsdatum :

    11.10.2023


    Format / Umfang :

    3271558 byte




    Medientyp :

    Aufsatz (Konferenz)


    Format :

    Elektronische Ressource


    Sprache :

    Englisch



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

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


    UAV target detection algorithm based on improved YOLOv5s

    Zhang, Tao / Wang, Fenmei / Chen, Dongxu et al. | SPIE | 2023



    Tank Armored Vehicle Target Detection Based on Improved YOLOv5s

    Li, Xinwei / Mao, Yuxin / Yu, Jianjun et al. | IEEE | 2024


    An Improved YOLOv5-Based Small Target Detection Method for UAV Aerial Image

    Li, Ruoyu / Gao, Yang / Zhang, Ruixing | Springer Verlag | 2023