IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v12y2024i11p1728-d1407276.html
   My bibliography  Save this article

CSD-YOLO: A Ship Detection Algorithm Based on a Deformable Large Kernel Attention Mechanism

Author

Listed:
  • Tao Wang

    (School of Shipping and Marine Engineering, Chongqing Jiaotong University, Chongqing 400074, China)

  • Han Zhang

    (School of Tourism and Media, Chongqing Jiaotong University, Chongqing 400074, China)

  • Dan Jiang

    (School of Shipping and Marine Engineering, Chongqing Jiaotong University, Chongqing 400074, China)

Abstract

Ship detection and identification play pivotal roles in ensuring navigation safety and facilitating efficient maritime traffic management. Aiming at ship detection in complex environments, which often faces problems such as the dense occlusion of ship targets, low detection accuracy, and variable environmental conditions, in this paper, we propose a ship detection algorithm CSD-YOLO (Context guided block module, Slim-neck, Deformable large kernel attention-You Only Look Once) based on the deformable large kernel attention (D-LKA) mechanism, which was improved based on YOLOv8 to enhance its performance. This approach integrates several innovations to bolster its performance. Initially, the utilization of the Context Guided Block module (CG block) enhanced the c2f module of the backbone network, thereby augmenting the feature extraction capabilities and enabling a more precise capture of the key image information. Subsequently, the introduction of a novel neck architecture and the incorporation of the slim-neck module facilitated more effective feature fusion, thereby enhancing both the accuracy and efficiency of detection. Furthermore, the algorithm incorporates a D-LKA mechanism to dynamically adjust the convolution kernel shape and size, thereby enhancing the model’s adaptability to varying ship target shapes and sizes. To address data scarcity in complex marine environments, the experiments utilized a fused dataset comprising the SeaShips dataset and a proprietary dataset. The experimental results demonstrate that the CSD-YOLO algorithm outperformed the YOLOv8n algorithm across all model evaluation metrics. Specifically, the precision rate (precision) was 91.5%, the recall rate (recall) was 89.5%, and the mean accuracy (mAP) was 91.5%. Compared to the benchmark algorithm, the Recall was improved by 0.7% and the mAP was improved by 0.4%. These results indicate that the CSD-YOLO algorithm can effectively meet the requirements for ship target recognition and tracking in complex marine environments.

Suggested Citation

  • Tao Wang & Han Zhang & Dan Jiang, 2024. "CSD-YOLO: A Ship Detection Algorithm Based on a Deformable Large Kernel Attention Mechanism," Mathematics, MDPI, vol. 12(11), pages 1-19, June.
  • Handle: RePEc:gam:jmathe:v:12:y:2024:i:11:p:1728-:d:1407276
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/12/11/1728/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/12/11/1728/
    Download Restriction: no
    ---><---

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:12:y:2024:i:11:p:1728-:d:1407276. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.