Abstract
At present, two main problems, which are the multi-scale of ship targets and the lightweight of detection models, restrict the real-time and on-orbit detection of ship targets on SAR images. To solve two problems, we propose a lightweight ship detection network (LS-YOLO) based on YOLOv5 model for SAR images. In the proposed network, we propose two modules, namely, Feature Refinement Module (FRM) and DCSP. The FRM module is designed to solve the multi-scale problem of ship targets in SAR images. This structure can effectively expand the receptive field of the model and improve the detection ability of small target ships. DCSP is lightweight module based on YOLOv5 CSP. This module effectively reduces model parameters and computation while keeping feature extraction ability as much as possible. The LS-YOLO detection speed is up to 1.2 ms, the accuracy (AP) is 96.6%, and the model size is only 3.8 MB. It can balance detection accuracy and detection speed, and provide reference for the construction of real-time detection network.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Li, D., Liang, Q., Liu, H., Liu, Q., Liu, H., Liao, G.: A novel multidimensional domain deep learning network for SAR ship detection. IEEE Trans. Geosci. Remote Sens. 60, 1–13 (2021)
Wu, Z., Hou, B., Jiao, L.: Multiscale CNN with autoencoder regularization joint contextual attention network for SAR image classification. IEEE Trans. Geosci. Remote Sens. 59(2), 1200–1213 (2020)
Lan, D.U., Wang, Z.C., Wang, Y., Wei, D., Lu, L.I.: Survey of research progress on target detection and discrimination of single-channel SAR images for complex scenes. Radars 9(1), 34–54 (2020)
Saepuloh, A., Bakker, E., Suminar, W.: The significance of SAR remote sensing in volcano-geology for hazard and resource potential mapping. In: Proceedings of the AIP Conference Proceedings, article no. 070005 (2017)
Schumacher, R., Schiller, J.: Non-cooperative target identification of battlefield targets-classification results based on SAR images. In: Proceedings of the IEEE International Radar Conference, pp. 167–172 (2005)
Xu, X., Zhang, X., Zhang, T.: Multi-scale SAR ship classification with convolutional neural network. In: Proceedings of the International Geoscience and Remote Sensing Symposium, pp. 4284–4287 (2021)
Yang, Y., Liao, Y., Ni, S., Lin, C.: Study of algorithm for aerial target detection based on lightweight neural network. In: Proceedings of the International Conference on Consumer Electronics and Computer Engineering, pp. 422–426 (2021)
Wang, C., Bi, F., Zhang, W., Chen, L.: An intensity-space domain CFAR method for ship detection in HR SAR images. IEEE Geosci. Remote Sens. Lett. 14(4), 529–533 (2017)
Pappas, O., Achim, A., Bull, D.: Superpixel-level CFAR detectors for ship detection in SAR imagery. IEEE Geosci. Remote Sens. Lett. 15(9), 1397–1401 (2018)
Shi, Z., Yu, X., Jiang, Z., Li, B.: Ship detection in high-resolution optical imagery based on anomaly detector and local shape feature. IEEE Trans. Geosci. Remote Sens. 52(8), 4511–4523 (2013)
Redmon, J., Divvala, S., Girshick, R., Farhadi, A.: You only look once: unified, real-time object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 779–788 (2016)
Redmon, J., Farhadi, A.: YOLO9000: better, faster, stronger. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7263–7271 (2017)
Redmon, J., Farhadi, A.: Yolov3: an incremental improvement. arXiv preprint arXiv: 1804.02767 (2018)
Liu, W., et al.: SSD: single shot multibox detector. In: Leibe, B., Matas, J., Sebe, N., Welling, M. (eds.) Computer Vision – ECCV 2016. ECCV 2016. Lecture Notes in Computer Science(), vol. 9905, pp. 21–37. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-46448-0_2
Girshick, R., Donahue, J., Darrell, T., Malik, J.: Rich feature hierarchies for accurate object detection and semantic segmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 580–587 (2014)
Girshick, R.: Fast R-CNN. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 1440–1448 (2015)
Chen, C., He, C., Hu, C., Pei, H., Jiao, L.: A deep neural network based on an attention mechanism for SAR ship detection in multiscale and complex scenarios. IEEE Access 7, 104848–104863 (2019)
Cui, Z., Li, Q., Cao, Z., Liu, N.: Dense attention pyramid networks for multi-scale ship detection in SAR images. IEEE Trans. Geosci. Remote Sens. 57(11), 8983–8997 (2019)
Zhang, R., Yao, J., Zhang, K., Feng, C., Zhang, J.: S-CNN-based ship detection from high-resolution remote sensing images. In: International Archives of the Photogrammetry, Remote Sensing Spatial Information Sciences 41 (2016)
Fu, C. Y., Liu, W., Ranga, A., Tyagi, A., Berg, A.C.: DSSD: deconvolutional single shot detector. arXiv preprint arXiv:1701.06659 (2017)
Wang, K., Liew, J. H., Zou, Y., Zhou, D., Feng, J.: Panet: few-shot image semantic segmentation with prototype alignment. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 9197–9206 (2019)
Tang, G., Zhuge, Y., Claramunt, C., Men, S.: N-Yolo: a SAR ship detection using noise-classifying and complete-target extraction. Remote Sens. 13(5), 871 (2021)
Zhou, K., Zhang, M., Wang, H., Tan, J.: Ship detection in SAR images based on multi-scale feature extraction and adaptive feature fusion. Remote Sens. 14(3), 755 (2022)
Acknowledgments
This work was supported in part by the National Science Foundation of China (Grant Nos. 61873335, 61833011); the Project of Science and Technology Commission of Shanghai Municipality, China (Grant Nos. 20ZR1420200, 21SQBS01600, 22JC1401400, 19510750300, 21190780300); and the 111 Project, China under Grant No. D18003.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
He, Y., Li, ZX., Wang, YL. (2022). LS-YOLO: Lightweight SAR Ship Targets Detection Based on Improved YOLOv5. In: Fang, L., Povey, D., Zhai, G., Mei, T., Wang, R. (eds) Artificial Intelligence. CICAI 2022. Lecture Notes in Computer Science(), vol 13606. Springer, Cham. https://doi.org/10.1007/978-3-031-20503-3_6
Download citation
DOI: https://doi.org/10.1007/978-3-031-20503-3_6
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-20502-6
Online ISBN: 978-3-031-20503-3
eBook Packages: Computer ScienceComputer Science (R0)