Skip to main content
Log in

A real-time deep learning network for ship detection in SAR images

  • Original Paper
  • Published:
Signal, Image and Video Processing Aims and scope Submit manuscript

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.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6

Similar content being viewed by others

Data availability

The datasets used or analyzed in the present study are accessible on the Internet.

References

  1. Zhang, T., Zhang, X.: A polarization fusion network with geometric feature embedding for SAR ship classification. Pattern Recogn. 123, 108365 (2022)

    Article  Google Scholar 

  2. Xu, X., Zhang, X., Zhang, T.: Multi-scale SAR ship classification with convolutional neural network. In: IGARSS, pp. 4284–4287 (2021)

  3. 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)

    Article  ADS  Google Scholar 

  4. 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)

    Article  ADS  Google Scholar 

  5. 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)

  6. 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)

    Article  ADS  Google Scholar 

  7. 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)

    Google Scholar 

  8. 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)

    Article  ADS  Google Scholar 

  9. 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)

    Article  ADS  Google Scholar 

  10. Zhang, T., Zhang, X., Ke, X.: Quad-fpn: a novel quad feature pyramid network for SAR ship detection. Remote Sensing 13(14), 2771 (2021)

    Article  ADS  Google Scholar 

  11. Girshick, R., Donahue, J., Darrell, T., Malik, J.: Rich feature hierarchies for accurate object detection and semantic segmentation. In: CVPR, pp. 580–587 (2014)

  12. Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You only look once: unified, real-time object detection. In: CVPR, pp. 779–788 (2016)

  13. Redmon, J., Farhadi, A.: YOLO9000: better, faster, stronger, pp. 6517–6525 (2017)

  14. Bochkovskiy, A., Wang, C.-Y., Liao, H.-Y.M.: YOLOv4: optimal speed and accuracy of object detection arXiv:2004.10934 (2020)

  15. Ge, Z., Liu, S., Wang, F., Li, Z., Sun, J.: Yolox: exceeding yolo series in 2021. arXiv preprint arXiv:2107.08430 (2021)

  16. RangiLyu: Nanodet. https://github.com/RangiLyu/nanodet

  17. Ma, N., Zhang, X., Zheng, H.-T., Sun, J.: Shufflenet v2: practical guidelines for efficient CNN architecture design. In: ECCV, pp. 116–131 (2018)

  18. ultralytics: Yolov5. https://github.com/ultralytics/yolov5

  19. Han, K., Wang, Y., Tian, Q., Guo, J., Xu, C., Xu, C.: Ghostnet: more features from cheap operations. In: CVPR, pp. 1580–1589 (2020)

  20. Hu, J., Shen, L., Sun, G.: Squeeze-and-excitation networks, pp. 7132–7141 (2018)

  21. Lee, Y., Park, J.: CenterMask: real-time anchor-free instance segmentation, pp. 13903–13912 (2020)

  22. 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)

  23. Song, G., Liu, Y., Wang, X.: Revisiting the sibling head in object detector. In: CVPR, pp. 11560–11569 (2020)

  24. Feng, C., Zhong, Y., Gao, Y., Scott, M.R., Huang, W.: TOOD: task-aligned one-stage object detection, pp. 3490–3499 (2021)

  25. 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)

  26. 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)

  27. 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)

  28. 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)

  29. Molchanov, P., Tyree, S., Karras, T., Aila, T., Kautz, J.: Pruning convolutional neural networks for resource efficient transfer learning arXiv:1611.06440 (2016)

  30. 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)

    Article  ADS  Google Scholar 

  31. 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)

    Article  ADS  Google Scholar 

Download references

Funding

No funding was received for conducting this study.

Author information

Authors and Affiliations

Authors

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

Correspondence to Huachun Zhang.

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.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

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

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11760-023-02892-6

Keywords

Navigation