MSFA-YOLO: A Multi-Scale SAR Ship Detection Algorithm Based on Fused Attention

被引:7
|
作者
Zhao, Liangjun [1 ]
Ning, Feng [2 ]
Xi, Yubin [1 ]
Liang, Gang [1 ]
He, Zhongliang [1 ]
Zhang, Yuanyang [1 ]
机构
[1] Sichuan Univ Sci & Engn, Sch Comp Sci & Engn, Yibin 644000, Peoples R China
[2] Sichuan Univ Sci & Engn, Sch Automat & Informat Engn, Yibin 644000, Peoples R China
关键词
Ship detection; SAR image; YOLO;
D O I
10.1109/ACCESS.2024.3365777
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Leveraging the excellent feature representation capabilities of neural networks, deep learning methods have been widely adopted for object detection in synthetic aperture radar (SAR) images. However, persistent challenges are encountered in SAR ship detection due to factors such as small ship sizes, high noise levels, multiple targets, and scale variations. To address these complexities, in this paper, the MSFA-YOLO algorithm, a novel multiscale SAR ship detection approach em-powered by a fused attention mechanism, is presented. The proposed algorithm incorporates several key enhancements. The fused attention c2fSE module is integrated into the YOLOv8n baseline network to optimize feature extraction for SAR ships. In addition, the DenseASPP module is incorporated to enhance the model's adaptability to ships of varying scales, improving its ca-pability to accommodate larger ships within lower model scales. Furthermore, the Wise-IoU loss function is adopted, and a dynamic non-monotonic focusing mechanism is employed for bounding box loss, significantly enhancing the model's ability to handle low-quality images. Extensive experiments conducted on benchmark datasets, namely SAR-Ship-Dataset, SSDD, and HRSID, validate the robustness and reliability of the proposed model. Experimental results demonstrate significant performance improvements over YOLOv8n: a 3.1% enhancement in mAP75 and a 2.1% boost in mAP50-95 on the SAR-Ship-Dataset, a 0.7% increase in mAP75 and a 0.5% increase in mAP50-95 on the SSDD dataset, and a 1.8% increase in mAP75 and a 0.7% increase in mAP50-95 on the HRSID dataset. Exhibiting superior performance to existing SAR ship detection models in terms of accuracy, the MSFA-YOLO algorithm represents a significant advancement, establishing itself as the current state-of-the-art algorithm in SAR ship detection.
引用
收藏
页码:24554 / 24568
页数:15
相关论文
共 50 条
  • [1] LMSD-YOLO: A Lightweight YOLO Algorithm for Multi-Scale SAR Ship Detection
    Guo, Yue
    Chen, Shiqi
    Zhan, Ronghui
    Wang, Wei
    Zhang, Jun
    REMOTE SENSING, 2022, 14 (19)
  • [2] DB-YOLO: A Duplicate Bilateral YOLO Network for Multi-Scale Ship Detection in SAR Images
    Zhu, Haozhen
    Xie, Yao
    Huang, Huihui
    Jing, Chen
    Rong, Yingjiao
    Wang, Changyuan
    SENSORS, 2021, 21 (23)
  • [3] FS-YOLO: a multi-scale SAR ship detection network in complex scenes
    Cai, Shouwen
    Meng, Hao
    Yuan, Ming
    Wu, Junbao
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (6-7) : 5017 - 5027
  • [4] FastPFM: a multi-scale ship detection algorithm for complex scenes based on SAR images
    Wang, Wei
    Han, Dezhi
    Chen, Chongqing
    Wu, Zhongdai
    CONNECTION SCIENCE, 2024, 36 (01)
  • [5] Dense Attention Pyramid Networks for Multi-Scale Ship Detection in SAR Images
    Cui, Zongyong
    Li, Qi
    Cao, Zongjie
    Liu, Nengyuan
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (11): : 8983 - 8997
  • [6] Electrode defect YOLO detection algorithm based on attention mechanism and multi-scale feature fusion
    Li Y.-W.
    Sun H.-R.
    Hu Y.-M.
    Han Y.-J.
    Kongzhi yu Juece/Control and Decision, 2023, 38 (09): : 2578 - 2586
  • [7] Multi-Scale Ship Detection Algorithm Based on a Lightweight Neural Network for Spaceborne SAR Images
    Liu, Shanwei
    Kong, Weimin
    Chen, Xingfeng
    Xu, Mingming
    Yasir, Muhammad
    Zhao, Limin
    Li, Jiaguo
    REMOTE SENSING, 2022, 14 (05)
  • [8] Lightweight algorithm for multi-scale ship detection based on high-resolution SAR images
    Kong, Weimin
    Liu, Shanwei
    Xu, Mingming
    Yasir, Muhammad
    Wang, Dawei
    Liu, Wantao
    INTERNATIONAL JOURNAL OF REMOTE SENSING, 2023, 44 (04) : 1390 - 1415
  • [9] DCAPT-Net: A Multi-Scale Ship Detection Algorithm for SAR Images
    Song, Haosheng
    Wang, Lin
    Mi, Luran
    Yu, Hua
    Can, Jingwei
    2024 10TH INTERNATIONAL CONFERENCE ON BIG DATA AND INFORMATION ANALYTICS, BIGDIA 2024, 2024, : 441 - 446
  • [10] SAR ship detection based on adaptive anchor and multi-scale enhancement
    Shao Z.
    Zhang X.
    Zhang T.
    Zeng T.
    Xi Tong Gong Cheng Yu Dian Zi Ji Shu/Systems Engineering and Electronics, 2024, 46 (04): : 1204 - 1211