Ship Detection in Synthetic Aperture Radar Images Based on BiLevel Spatial Attention and Deep Poly Kernel Network
-
Published:2024-08-12
Issue:8
Volume:12
Page:1379
-
ISSN:2077-1312
-
Container-title:Journal of Marine Science and Engineering
-
language:en
-
Short-container-title:JMSE
Author:
Tian Siyuan1ORCID, Jin Guodong1, Gao Jing1, Tan Lining1, Xue Yuanliang1, Li Yang1, Liu Yantong2ORCID
Affiliation:
1. Xi’an Research Institute of Hi-Tech, Xi’an 710025, China 2. Department of Computer and Information Engineering, Kunsan National University, Gunsan 54150, Republic of Korea
Abstract
Synthetic aperture radar (SAR) is a technique widely used in the field of ship detection. However, due to the high ship density, fore-ground-background imbalance, and varying target sizes, achieving lightweight and high-precision multiscale ship object detection remains a significant challenge. In response to these challenges, this research presents YOLO-MSD, a multiscale SAR ship detection method. Firstly, we propose a Deep Poly Kernel Backbone Network (DPK-Net) that utilizes the Optimized Convolution (OC) Module to reduce data redundancy and the Poly Kernel (PK) Module to improve the feature extraction capability and scale adaptability. Secondly, we design a BiLevel Spatial Attention Module (BSAM), which consists of the BiLevel Routing Attention (BRA) and the Spatial Attention Module. The BRA is first utilized to capture global information. Then, the Spatial Attention Module is used to improve the network’s ability to localize the target and capture high-quality detailed information. Finally, we adopt a Powerful-IoU (P-IoU) loss function, which can adjust to the ship size adaptively, effectively guiding the anchor box to achieve faster and more accurate detection. Using HRSID and SSDD as experimental datasets, mAP of 90.2% and 98.8% are achieved, respectively, outperforming the baseline by 5.9% and 6.2% with a model size of 12.3 M. Furthermore, the network exhibits excellent performance across various ship scales.
Funder
National Natural Science Foundation of China
Reference50 articles.
1. Dudczyk, J., and Rybak, Ł. (2023). Application of Data Particle Geometrical Divide Algorithms in the Process of Radar Signal Recognition. Sensors, 23. 2. Li, J., Xu, C., Su, H., Gao, L., and Wang, T. (2022). Deep Learning for SAR Ship Detection: Past, Present and Future. Remote Sens., 14. 3. Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C.-Y., and Berg, A.C. (2016, January 11–14). SSD: Single shot multibox detector. Proceedings of the Computer Vision–ECCV 2016: 14th European Conference, Amsterdam, The Netherlands. 4. Bochkovskiy, A., Wang, C.-Y., and Liao, H.-Y.M. (2020). YOLOV4: Optimal speed and accuracy of object detection. arXiv. 5. Ge, Z., Liu, S., Wang, F., Li, Z., and Sun, J. (2021). YOLOX: Exceeding YOLO series in 2021. arXiv.
|
|