Due to the unique distance and angles involved in satellite remote sensing, ships appear with a small pixel area in images, leading to insufficient feature representation. This results in suboptimal performance in ship detection, including potential misses and false detections. Moreover, the complexity of backgrounds in remote sensing images of ships and the clustering of vessels also adversely affect the accuracy of ship detection. Therefore, this paper proposes an optimized model named SSMA-YOLO, based on YOLOv8n. First, this paper introduces a newly designed SSC2f structure that incorporates spatial and channel convolution (SCConv) and spatial group-wise enhancement (SGE) attention mechanisms. This design reduces spatial and channel redundancies within the neural network, enhancing detection accuracy while simultaneously reducing the model’s parameter count. Second, the newly designed MC2f structure employs the multidimensional collaborative attention (MCA) mechanism to efficiently model spatial and channel features, enhancing recognition efficiency in complex backgrounds. Additionally, the asymptotic feature pyramid network (AFPN) structure was designed for progressively fusing multi-level features from the backbone layers, overcoming challenges posed by multi-scale variations. Experiments of the ships dataset show that the proposed model achieved a 4.4% increase in mAP compared to the state-of-the-art single-stage target detection YOLOv8n model while also reducing the number of parameters by 23%.


    Zugriff

    Download


    Exportieren, teilen und zitieren



    Titel :

    SSMA-YOLO: A Lightweight YOLO Model with Enhanced Feature Extraction and Fusion Capabilities for Drone-Aerial Ship Image Detection


    Beteiligte:
    Yuhang Han (Autor:in) / Jizhuang Guo (Autor:in) / Haoze Yang (Autor:in) / Renxiang Guan (Autor:in) / Tianjiao Zhang (Autor:in)


    Erscheinungsdatum :

    2024




    Medientyp :

    Aufsatz (Zeitschrift)


    Format :

    Elektronische Ressource


    Sprache :

    Unbekannt





    Lightweight ship detection method based on YOLO-FNC model

    Bingyan ZHANG / Chuang ZHANG / Zhennan SHI et al. | DOAJ | 2024

    Freier Zugriff


    Performance Comparison of YOLO Algorithms in Drone Detection

    Jain, Rakshit / Shrivastav, Sujal / Kakde, Sumit et al. | IEEE | 2024


    Ship detection based on improved YOLO algorithm

    Zhang, Xiaoqing / Zhang, Zhiqiang | IEEE | 2023