Abstract
In the field of ship detection in synthetic aperture radar (SAR) images using deep learning, traditional models often face challenges related to their complex structure and high computational demands. To address these problems, this paper proposes a lightweight end-to-end convolutional neural network called BGD-Net, which is based on anchor-free methods. This network incorporates a novel feature pyramid called Ghost, Cross-Stage-Partial, and Path Aggregation Network (G-CSP-PAN) to enhance the detection performance across targets of varying scales. Additionally, it introduces an efficient decoupled detection head, termed the efficient decoupled head (ED-Head), to enhance the interaction between regression and classification. Furthermore, an optimized loss function named optimized efficient intersection over union (OEIoU) loss is proposed for edge regression. The proposed method is evaluated on public datasets, SSDD and SAR-Ship-Dataset, demonstrating a balance between detection accuracy and efficiency.
Similar content being viewed by others
Data availability
The datasets used or analyzed in the present study are accessible on the Internet.
References
Zhang, T., Zhang, X.: A polarization fusion network with geometric feature embedding for SAR ship classification. Pattern Recogn. 123, 108365 (2022)
Xu, X., Zhang, X., Zhang, T.: Multi-scale SAR ship classification with convolutional neural network. In: IGARSS, pp. 4284–4287 (2021)
Xiong, B., Sun, Z., Wang, J., Leng, X., Ji, K.: A lightweight model for ship detection and recognition in complex-scene SAR images. Remote Sensing 14(23), 6053 (2022)
Zhang, T., Zhang, X., Liu, C., Shi, J., Wei, S., Ahmad, I., Zhan, X., Zhou, Y., Pan, D., Li, J., et al.: Balance learning for ship detection from synthetic aperture radar remote sensing imagery. ISPRS J. Photogramm. Remote. Sens. 182, 190–207 (2021)
Ke, X., Zhang, X., Zhang, T., Shi, J., Wei, S.: Sar ship detection based on swin transformer and feature enhancement feature pyramid network. In: IGARSS, pp. 2163–2166 (2022)
Sun, Z., Leng, X., Lei, Y., Xiong, B., Ji, K., Kuang, G.: Bifa-yolo: a novel yolo-based method for arbitrary-oriented ship detection in high-resolution SAR images. Remote Sensing 13(21), 4209 (2021)
Zhang, T., Zhang, X.: A mask attention interaction and scale enhancement network for SAR ship instance segmentation. IEEE Geosci. Remote Sens. Lett. 19, 1–5 (2022)
Zhang, T., Zhang, X., Shi, J., Wei, S.: Depthwise separable convolution neural network for high-speed SAR ship detection. Remote Sensing 11(21), 2483 (2019)
Zhang, T., Zhang, X., Shi, J., Wei, S.: Hyperli-net: a hyper-light deep learning network for high-accurate and high-speed ship detection from synthetic aperture radar imagery. ISPRS J. Photogramm. Remote. Sens. 167, 123–153 (2020)
Zhang, T., Zhang, X., Ke, X.: Quad-fpn: a novel quad feature pyramid network for SAR ship detection. Remote Sensing 13(14), 2771 (2021)
Girshick, R., Donahue, J., Darrell, T., Malik, J.: Rich feature hierarchies for accurate object detection and semantic segmentation. In: CVPR, pp. 580–587 (2014)
Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You only look once: unified, real-time object detection. In: CVPR, pp. 779–788 (2016)
Redmon, J., Farhadi, A.: YOLO9000: better, faster, stronger, pp. 6517–6525 (2017)
Bochkovskiy, A., Wang, C.-Y., Liao, H.-Y.M.: YOLOv4: optimal speed and accuracy of object detection arXiv:2004.10934 (2020)
Ge, Z., Liu, S., Wang, F., Li, Z., Sun, J.: Yolox: exceeding yolo series in 2021. arXiv preprint arXiv:2107.08430 (2021)
RangiLyu: Nanodet. https://github.com/RangiLyu/nanodet
Ma, N., Zhang, X., Zheng, H.-T., Sun, J.: Shufflenet v2: practical guidelines for efficient CNN architecture design. In: ECCV, pp. 116–131 (2018)
ultralytics: Yolov5. https://github.com/ultralytics/yolov5
Han, K., Wang, Y., Tian, Q., Guo, J., Xu, C., Xu, C.: Ghostnet: more features from cheap operations. In: CVPR, pp. 1580–1589 (2020)
Hu, J., Shen, L., Sun, G.: Squeeze-and-excitation networks, pp. 7132–7141 (2018)
Lee, Y., Park, J.: CenterMask: real-time anchor-free instance segmentation, pp. 13903–13912 (2020)
Li, D., Hu, J., Wang, C., Li, X., She, Q., Zhu, L., Zhang, T., Chen, Q.: Involution: inverting the inherence of convolution for visual recognition. In: CVPR, pp. 12316–12325 (2021)
Song, G., Liu, Y., Wang, X.: Revisiting the sibling head in object detector. In: CVPR, pp. 11560–11569 (2020)
Feng, C., Zhong, Y., Gao, Y., Scott, M.R., Huang, W.: TOOD: task-aligned one-stage object detection, pp. 3490–3499 (2021)
Xu, S., Wang, X., Lv, W., Chang, Q., Cui, C., Deng, K., Wang, G., Dang, Q., Wei, S., Du, Y., Lai, B.: PP-YOLOE: an evolved version of YOLO. arXiv:2203.16250 (2022)
Rezatofighi, S.H., Tsoi, N., Gwak, J., Sadeghian, A., Reid, I.D., Savarese, S.: Generalized intersection over union: a metric and a loss for bounding box regression, pp. 658–666 (2019)
Zhang, Y.-F., Ren, W., Zhang, Z., Jia, Z., Wang, L., Tan, T.: Focal and efficient IOU loss for accurate bounding box regression arXiv:2101.08158 (2021)
Zhang, T., Zhang, X., Li, J., Xu, X., Wang, B., Zhan, X., Xu, Y., Ke, X., Zeng, T., Su, H., Ahmad, I., Pan, D., Liu, C., Zhou, Y., Shi, J., Wei, S.: SAR ship detection dataset (SSDD): official release and comprehensive data analysis 13(18) (2021)
Molchanov, P., Tyree, S., Karras, T., Aila, T., Kautz, J.: Pruning convolutional neural networks for resource efficient transfer learning arXiv:1611.06440 (2016)
Wang, Y., Wang, C., Zhang, H., Dong, Y., Wei, S.: A SAR dataset of ship detection for deep learning under complex backgrounds. Remote Sensing 11(7), 765 (2019)
Zhang, T., Zhang, X., Ke, X., Zhan, X., Shi, J., Wei, S., Pan, D., Li, J., Su, H., Zhou, Y., et al.: Ls-ssdd-v1. 0: a deep learning dataset dedicated to small ship detection from large-scale sentinel-1 SAR images. Remote Sensing 12(18), 2997 (2020)
Funding
No funding was received for conducting this study.
Author information
Authors and Affiliations
Contributions
W. X. Zhou was involved in conceptualization, investigation, software, validation, writing—original draft, and visualization. W. X. Zhou and H. C. Zhang were involved in methodology. H. C. Zhang was involved in resources, supervision, and writing—review and editing.
Corresponding author
Ethics declarations
Ethical approval
Not applicable.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Zhou, W., Zhang, H. A real-time deep learning network for ship detection in SAR images. SIViP 18, 1893–1899 (2024). https://doi.org/10.1007/s11760-023-02892-6
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11760-023-02892-6