Skip to main content
Log in

Lightweight SAR ship detection algorithm based on attention mechanism

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

Abstract

In recent years, with the rapid development of deep learning, the application in the field of synthetic aperture radar (SAR) image ship detection has received increasing attention. Most of the existing algorithms suffer from the problems of more network parameters and larger computation, so they are usually difficult to be deployed on embedded devices with limited storage and computational resources. The lightweight detection network cannot achieve good detection accuracy in complex scenes due to its small number of parameters and insufficient feature extraction and fusion for small-sized ships. Aiming at the above problems, in this paper, based on the YOLOv5 detection algorithm, the backbone and neck networks are designed to be lightweight, and the improved spatial feature pyramid module SimSPPFC is proposed to replace the SPPF module in YOLOv5, and the convolutional block attention module (CBAM) is used to construct an efficient feature extraction module to improve the feature extraction and selection capability of the network, and the weighted average introduced to the weighted non-maximum suppression (NMS) is introduced to replace the traditional NMS as the post-processing algorithm to further improve the localization accuracy of the prediction box. Experiments show that the proposed algorithm achieves 97.7% AP50 and 66.8% AP50:95 on the SSDD dataset with 0.438 M number of parameters and 1.1GFLOPs computational complexity. The algorithm takes into account both the detection accuracy and efficiency, and realizes a good trade-off between algorithmic lightweighting and detection accuracy.

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

Access this article

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

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

Similar content being viewed by others

Data availability

No datasets were generated or analysed during the current study.

References

  1. Liu, S., Kong, W., Chen, X., Xu, M., Yasir, M., Zhao, L., Li, J.: Multi-scale Ship Detection Algorithm based on a lightweight neural network for Spaceborne SAR images. Remote Sens. 14(5), 1149 (2022)

    Article  MATH  Google Scholar 

  2. Ao, W., Xu, F., Li, Y., Wang, H.: Detection and discrimination of ship targets in complex background from spaceborne ALOS-2 SAR images. IEEE J. Sel. Top. Appl. Earth Observations Remote Sens. 11(2), 536–550 (2018)

    Article  MATH  Google Scholar 

  3. Leng, X., Ji, K., Yang, K., Zou, H.: A bilateral CFAR algorithm for ship detection in SAR images. IEEE Geosci. Remote Sens. Lett. 12(7), 1536–1540 (2015)

    Article  MATH  Google Scholar 

  4. Ren, S., He, K., Girshick, R., & Sun, J: Faster R-CNN: Towards real-time object detection with region proposal networks. IEEE transactions on pattern analysis and machine intelligence. 39(6), 1137-1149 (2015)

  5. Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You only look once: Unified, real-time object detection. Proceedings of the IEEE conference on computer vision and pattern recognition. 779–788. (2016)

  6. Li, J., Qu, C., Shao, J.: Ship detection in SAR images based on an improved faster R-CNN. 2017 SAR in Big Data Era: Models, Methods and Applications (BIGSARDATA). IEEE, 1–6. (2017)

  7. Liu, N., Cao, Z., Cui, Z., Dang, S.: Multi-scale proposal generation for ship detection in SAR images. Remote Sens. 11(5), 526 (2019)

    Article  MATH  Google Scholar 

  8. Sun, Z., Dai, M., Leng, X., Lei, Y., Xiong, B., Ji, K., Kuang, G.: An anchor-free detection method for ship targets in high-resolution SAR images. IEEE J. Sel. Top. Appl. Earth Observations Remote Sens. 14, 7799–7816 (2021)

    Article  MATH  Google Scholar 

  9. Chen, S.Q., Zhan, R.H., Zhang, J.: Robust single stage detector based on two-stage regression for SAR ship detection. Proceedings of the 2nd International Conference on Innovation in Artificial Intelligence. 169–174. (2018)

  10. Zhao, S., Zhang, Z., Guo, W., Luo, Y.: An automatic ship detection method adapting to different satellites SAR images with feature alignment and compensation loss. IEEE Trans. Geosci. Remote Sens. 60, 1–17 (2022)

    Article  MATH  Google Scholar 

  11. Xie, F., Luo, H., Li, S., Liu, Y., Lin, B.: Using Clean Energy satellites to Interpret Imagery: A Satellite IoT oriented Lightweight object detection Framework for SAR Ship Detection. Sustainability. 14(15), 9277 (2022)

    Article  Google Scholar 

  12. Bai, L., Yao, C., Ye, Z., Xue, D., Lin, X., Hui, M.: Feature enhancement pyramid and shallow feature Reconstruction Network for SAR Ship Detection. IEEE J. Sel. Top. Appl. Earth Observations Remote Sens. 16, 1042–1056 (2023)

    Article  MATH  Google Scholar 

  13. Li, M., Wen, G., Huang, X., Li, K., Lin, S.: A lightweight detection model for SAR Aircraft in a Complex Environment. Remote Sens. 13(24), 5020 (2021)

    Article  MATH  Google Scholar 

  14. Miao, T., Zeng, H., Yang, W., Chu, B., Zou, F., Ren, W., Chen, J.: An improved lightweight RetinaNet for ship detection in SAR image. IEEE J. Sel. Top. Appl. Earth Observations Remote Sens. 15, 4667–4679 (2022)

    Article  Google Scholar 

  15. Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, Z., Jones, L., Gomez, A.N., Kaiser, L., Polosukhin, L.: Attention is all you need. Advances in Neural Information Processing Systems. 30, 5998–6008 (2017)

  16. Jie, H., Li, S., Gang, S.: Squeeze-and-excitation networks. Proceedings of the IEEE conference on computer vision and pattern recognition. 7132–7141. (2018)

  17. Woo, S., Park, J., Lee, J.Y., Kweon, I.S.: Cbam: Convolutional block attention module. Proceedings of the European conference on computer vision (ECCV). 3–19. (2018)

  18. Ultralytics Yolov5: Available online: (2022). https://github.com/ultralytics/yolov5 (accessed on 2 February 2022)

  19. Ning, C., Zhou, H., Song, Y., Tang, J.: Inception Single Shot MultiBox Detector for object detection. 2017 IEEE International Conference on Multimedia & Expo Workshops (ICMEW). IEEE, 549–554. (2017)

  20. Ioffe, S., Szegedy, C.: Batch normalization: Accelerating deep network training by reducing internal covariate shift. Int. Conf. Mach. Learn. pmlr, 448–456 (2015)

    Google Scholar 

  21. Hendrycks, D., Gimpel, K.: Gaussian error linear units (gelus). arXiv preprint arXiv:1606.08415. (2016)

  22. He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. Proceedings of the IEEE conference on computer vision and pattern recognition. 770–778. (2016)

  23. Bochkovskiy, A., Wang, C.Y., Liao, H.Y.M.: Yolov4: Optimal speed and accuracy of object detection. arXiv preprint arXiv:2004.10934. (2020)

  24. Zhang, H., Cisse, M., Dauphin, Y.N., Lopez-Paz, D.: mixup: Beyond empirical risk minimization. arXiv preprint arXiv:1710.09412. (2017)

  25. Lin, T.Y., Maire, M., Belongie, S., Hays, J., Perona, P., Ramanan, D., Dollar, P., Zitnick, C.L.: Microsoft coco: Common objects in context. Computer Vision–ECCV 2014: 13th European Conference, Zurich, Switzerland, September 6–12, 2014, Proceedings, Part V 13. Springer International Publishing, 740–755. (2014)

  26. Wang, C.Y., Bochkovskiy, A., Liao, H.Y.M.: YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 7464–7475. (2023)

  27. Guo, Y., Chen, S., Zhan, R., Wang, W., Zhang, J.: Sar ship detection based on yolov5 using cbam and bifpn. IGARSS 2022–2022 IEEE International Geoscience and Remote Sensing Symposium. IEEE, 2147–2150. (2022)

  28. Sandler, M., Howard, A., Zhu, M., Zhmoginov, A., Chen, L.C.: Mobilenetv2: Inverted residuals and linear bottlenecks. Proceedings of the IEEE conference on computer vision and pattern recognition. 4510–4520. (2018)

  29. Howard, A., Sandler, M., Chu, G., et al.: Searching for mobilenetv3. Proceedings of the IEEE/CVF international conference on computer vision. 1314–1324. (2019)

  30. Ma, N., Zhang, X., Zheng, H.T., Sun, J.: Shufflenet v2: Practical guidelines for efficient cnn architecture design. Proceedings of the European conference on computer vision (ECCV). 116–131. (2018)

  31. Li, C., Li, L., Jiang, H., et al.: YOLOv6: A single-stage object detection framework for industrial applications. arXiv preprint arXiv:2209.02976. (2022)

  32. Xie, F., Luo, H., Li, S., et al.: Using clean energy satellites to interpret imagery: A satellite IoT oriented lightweight object detection framework for SAR ship detection. Sustainability. 14, 15: 9277 (2022)

    Article  MATH  Google Scholar 

  33. Guo, J., Wang, Z., Zhang, S.: Multi-scale ship detection in SAR images based on multiple attention cascade convolutional neural networks. 2020 international conference on virtual reality and intelligent systems (ICVRIS). 438–441. (2020)

  34. Luo, P., Wang, Z., Wang, T., et al.: SAR Ship Target Detection Algorithm Based on Improved YOLOv7. 2023 China Automation Congress (CAC). 2070–2074. (2023)

Download references

Acknowledgements

We gratefully acknowledge the anonymous reviewers who read the drafts and provided many helpful suggestions.

Funding

The authors declare that no funds, grants, or other support were received during the preparation of this manuscript.

Author information

Authors and Affiliations

Authors

Contributions

Weihong Fu and Peiyuan Zheng wrote the main manuscript text. Peiyuan Zheng prepared all the figures and Weihong Fu prepared the experiments to get the result. All authors reviewed the manuscript.

Corresponding author

Correspondence to Weihong Fu.

Ethics declarations

Competing interests

The authors declare no competing interests.

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

Fu, W., Zheng, P. Lightweight SAR ship detection algorithm based on attention mechanism. SIViP 19, 79 (2025). https://doi.org/10.1007/s11760-024-03641-z

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • DOI: https://doi.org/10.1007/s11760-024-03641-z

Keywords

Navigation