Classification of ships in synthetic aperture radar (SAR) images plays a critical role in maritime search and rescue as well as national defense security. Over the past few years, the swift advancement rapid of deep learning has opened up innovative technological directions for precise classification of SAR ship images. However, the unique imaging mechanism of SAR images, resulting in a lack of ship samples and insufficient feature information, continues to make accurate classification of SAR ships challenging. In this study, we propose an improved vision transformer (ViT) method using a full attention (FA) mechanism, incorporating channel attention after Self-Attention, dynamically selecting channels through reweighting to enhance comprehensive feature representation. To address the issue of insufficient training samples, we designed an image reconstruction module based on SRGAN, which generates high-resolution and detailed image data through adversarial learning, effectively compensating for the sample quantity in the dataset. Ultimately, the results section demonstrates that the method introduced delivers enhanced classification accuracy relative to current methods.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Ship Classification in SAR Images Based on Full-Attention ViT and Image Reconstruction

  • Lu Wang,
  • Chunhui Zhao,
  • Tong Wang,
  • Bailiang Sun,
  • Yuhang Qi,
  • Bin Qi,
  • Tomoaki Ohtsuki

摘要

Classification of ships in synthetic aperture radar (SAR) images plays a critical role in maritime search and rescue as well as national defense security. Over the past few years, the swift advancement rapid of deep learning has opened up innovative technological directions for precise classification of SAR ship images. However, the unique imaging mechanism of SAR images, resulting in a lack of ship samples and insufficient feature information, continues to make accurate classification of SAR ships challenging. In this study, we propose an improved vision transformer (ViT) method using a full attention (FA) mechanism, incorporating channel attention after Self-Attention, dynamically selecting channels through reweighting to enhance comprehensive feature representation. To address the issue of insufficient training samples, we designed an image reconstruction module based on SRGAN, which generates high-resolution and detailed image data through adversarial learning, effectively compensating for the sample quantity in the dataset. Ultimately, the results section demonstrates that the method introduced delivers enhanced classification accuracy relative to current methods.