Oriented Ship Detection in SAR Image with Data Augmentation Based on Swin Transformer

Chunyang Shi,Dunbin Shen,Xiaorui Ma,Hongyu Wang
DOI: https://doi.org/10.1109/yac63405.2024.10598538
2024-01-01
Abstract:In synthetic aperture radar (SAR) image, oriented ship detection can reveal the true shape and direction information of the ship for better identification, tracking, navigation and monitoring. However, the performance of existing oriented detectors based on convolutional neural networks (CNNs) is restricted due to the difficulty of capturing long-distance dependencies. In addition, there is an unbalanced distribution among the number of ship directions, easily leading to overfitting. In response to these problems, we propose a method of oriented ship detection in SAR image with data augmentation based on Swin Transformer. Firstly, the oriented data augmentation method is used to solve the problem of uneven distribution in ship directions, allowing the model to allocate equal effort in learning different kinds of orientation features. Secondly, the Swin Transformer is employed as the backbone to extract hierarchical features of SAR image to solve the problem that CNNs may be difficult to effectively capture global information, allowing the network to better cope with dense arrangement. Additionally, Gaussian distance loss is used to suppress the impact of scattering interference in SAR image. Experiments conducted on two benchmark datasets validate the effectiveness of the proposed method.
What problem does this paper attempt to address?