A synthetic aperture radar small ship detector based on transformers and multi-dimensional parallel feature extraction

被引:0
作者
Fu, Xinyi [1 ]
Zhou, Zhengchun [2 ]
Meng, Hua [1 ]
Li, Shuting [1 ]
机构
[1] Southwest Jiaotong Univ, Sch Math, Chengdu 611756, Peoples R China
[2] Southwest Jiaotong Univ, Sch Informat Sci & Technol, Chengdu 611756, Sichuan, Peoples R China
关键词
Feature enhancement; Transformer; Small ship detection; Noise interference; Synthetic aperture radar; SAR IMAGES; NETWORK;
D O I
10.1016/j.engappai.2024.109049
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Deep learning-based target detection methods have been widely used for Synthetic aperture radar (SAR) ship detection in recent years. However, the large amount of clutter noise and land interference information in SAR images poses challenges to small ship target detection. Existing methods are still inadequate for solving the problem of noise-induced feature loss of small targets, and feature extraction of small targets is inadequate, leading to poor detection accuracy. In response to these problems, we propose a new small ship detector-you only look once (SSD-YOLO). First, to solve the problem that noise interference leads to the loss of ship body features, a small target feature enhancement module (STFEM) based on multi-dimensional parallel feature extraction is introduced. STFEM uses attention mechanisms to capture the main features of small ships from multiple dimensions, effectively enhancing the boundary texture information of the ship body. In addition, a transformer-based skip connection path aggregation network (Tr-PANet) is introduced to adequately extract the contextual information of small targets. Tr-PANet uses self-attention to model the non-local contextual features of small targets in bottom-up feature fusion and uses skip connection to maximally maintain the global structural information of small targets. Since SSD-YOLO is based on a lightweight model, the number of parameters and the computational amount are lower than most of the comparative models. Moreover, SSD-YOLO shows the best detection performance on three real datasets.
引用
收藏
页数:15
相关论文
共 62 条
  • [1] A Novel Anchor-Free Detector Using Global Context-Guide Feature Balance Pyramid and United Attention for SAR Ship Detection
    Bai, Lin
    Yao, Cheng
    Ye, Zhen
    Xue, Dongling
    Lin, Xiangyuan
    Hui, Meng
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
  • [2] Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, DOI 10.48550/ARXIV.2004.10934, 10.48550/arXiv.2004.10934]
  • [3] Cascade R-CNN: Delving into High Quality Object Detection
    Cai, Zhaowei
    Vasconcelos, Nuno
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 6154 - 6162
  • [4] SPANet: A Self-Balancing Position Attention Network for Anchor-Free SAR Ship Detection
    Chang, Hao
    Fu, Xiongjun
    Lu, Jihua
    Guo, Kunyi
    Dong, Jian
    Zhao, Congxia
    Feng, Cheng
    Li, Ziying
    Zhang, Yue
    [J]. IEEE JOURNAL OF SELECTED TOPICS IN APPLIED EARTH OBSERVATIONS AND REMOTE SENSING, 2023, 16 : 8363 - 8378
  • [5] Towards Large-Scale Small Object Detection: Survey and Benchmarks
    Cheng, Gong
    Yuan, Xiang
    Yao, Xiwen
    Yan, Kebing
    Zeng, Qinghua
    Xie, Xingxing
    Han, Junwei
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2023, 45 (11) : 13467 - 13488
  • [6] Design of Customized Adaptive Radar Detectors in the CFAR Feature Plane
    Coluccia, Angelo
    Fascista, Alessio
    Ricci, Giuseppe
    [J]. IEEE TRANSACTIONS ON SIGNAL PROCESSING, 2022, 70 : 5133 - 5147
  • [7] Cui Y., 2011, P 3 AS PAC C SYNTH A, P1
  • [8] Ship Detection in Large-Scale SAR Images Via Spatial Shuffle-Group Enhance Attention
    Cui, Zongyong
    Wang, Xiaoya
    Liu, Nengyuan
    Cao, Zongjie
    Yang, Jianyu
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2021, 59 (01): : 379 - 391
  • [9] Dense Attention Pyramid Networks for Multi-Scale Ship Detection in SAR Images
    Cui, Zongyong
    Li, Qi
    Cao, Zongjie
    Liu, Nengyuan
    [J]. IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2019, 57 (11): : 8983 - 8997
  • [10] Deformable Convolutional Networks
    Dai, Jifeng
    Qi, Haozhi
    Xiong, Yuwen
    Li, Yi
    Zhang, Guodong
    Hu, Han
    Wei, Yichen
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 764 - 773