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.
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03641-z/MediaObjects/11760_2024_3641_Fig1_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03641-z/MediaObjects/11760_2024_3641_Fig2_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03641-z/MediaObjects/11760_2024_3641_Fig3_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03641-z/MediaObjects/11760_2024_3641_Fig4_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03641-z/MediaObjects/11760_2024_3641_Fig5_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03641-z/MediaObjects/11760_2024_3641_Fig6_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03641-z/MediaObjects/11760_2024_3641_Fig7_HTML.png)
![](http://media.springernature.com/m312/springer-static/image/art%3A10.1007%2Fs11760-024-03641-z/MediaObjects/11760_2024_3641_Fig8_HTML.png)
Similar content being viewed by others
Data availability
No datasets were generated or analysed during the current study.
References
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)
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)
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)
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)
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)
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)
Liu, N., Cao, Z., Cui, Z., Dang, S.: Multi-scale proposal generation for ship detection in SAR images. Remote Sens. 11(5), 526 (2019)
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)
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)
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)
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)
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)
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)
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)
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)
Jie, H., Li, S., Gang, S.: Squeeze-and-excitation networks. Proceedings of the IEEE conference on computer vision and pattern recognition. 7132–7141. (2018)
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)
Ultralytics Yolov5: Available online: (2022). https://github.com/ultralytics/yolov5 (accessed on 2 February 2022)
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)
Ioffe, S., Szegedy, C.: Batch normalization: Accelerating deep network training by reducing internal covariate shift. Int. Conf. Mach. Learn. pmlr, 448–456 (2015)
Hendrycks, D., Gimpel, K.: Gaussian error linear units (gelus). arXiv preprint arXiv:1606.08415. (2016)
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)
Bochkovskiy, A., Wang, C.Y., Liao, H.Y.M.: Yolov4: Optimal speed and accuracy of object detection. arXiv preprint arXiv:2004.10934. (2020)
Zhang, H., Cisse, M., Dauphin, Y.N., Lopez-Paz, D.: mixup: Beyond empirical risk minimization. arXiv preprint arXiv:1710.09412. (2017)
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)
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)
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)
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)
Howard, A., Sandler, M., Chu, G., et al.: Searching for mobilenetv3. Proceedings of the IEEE/CVF international conference on computer vision. 1314–1324. (2019)
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)
Li, C., Li, L., Jiang, H., et al.: YOLOv6: A single-stage object detection framework for industrial applications. arXiv preprint arXiv:2209.02976. (2022)
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)
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)
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)
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
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
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.
About this article
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
Received:
Revised:
Accepted:
Published:
DOI: https://doi.org/10.1007/s11760-024-03641-z