Abstract
Deep learning techniques have made significant advancements in computer vision. The YOLO algorithm, a representative single-stage detection approach, has demonstrated remarkable results in detecting ship targets in SAR images. We introduce an enhanced ship target detection model for SAR images, utilizing the improved YOLOv7 object detection network. We incorporate the coordinate attention mechanism into the network to enable automatic detection and diagnosis of ship targets within SAR images. To enhance the robustness and positioning accuracy of the detection network, we replace the CIoU regression loss in YOLOv7 with the SIoU loss, reducing the complexity of the loss function. Additionally, we integrate the rotating target detection technology into the network to mitigate the impact of target overlap on detection results. Comprehensive experiments conducted on the Capella Space synthetic aperture radar datasets validate that the proposed methodology achieves superior performance in multiple evaluation metrics, including precision, recall, and mean average precision.
Similar content being viewed by others
Availability of data and materials
The authors will supply the relevant data in response to reasonable requests.
Code availability
The authors will supply the relevant source code in response to reasonable requests.
References
Mondini AC, Guzzetti F, Chang K-T, Monserrat O, Martha TR, Manconi A (2021) Landslide failures detection and mapping using synthetic aperture radar: Past, present and future. Earth Sci Rev 216:103574
Hamidi E, Peter BG, Muñoz DF, Moftakhari H, Moradkhani H (2023) Fast flood extent monitoring with sar change detection using google earth engine. IEEE Trans Geosci Remote Sens 61:1–19
Zhou Z, Chen J, Huang Z, Lv J, Song J, Luo H, Wu B, Li Y, Diniz PS (2023) Hrle-sardet: a lightweight sar target detection algorithm based on hybrid representation learning enhancement. IEEE Trans Geosci Remote Sens 61:1–22
Wang X, Liu J, Liu X, Liu Z, Khalaf OI, Ji J, Ouyang Q (2022) Ship feature recognition methods for deep learning in complex marine environments. Complex Intell Syst 8(5):3881–3897
Cui Z, Wang X, Liu N, Cao Z, Yang J (2020) Ship detection in large-scale sar images via spatial shuffle-group enhance attention. IEEE Trans Geosci Remote Sens 59(1):379–391
Dargan S, Kumar M, Ayyagari MR, Kumar G (2020) A survey of deep learning and its applications: a new paradigm to machine learning. Arch Comput Methods Eng 27:1071–1092
Xu Y, Liu X, Cao X, Huang C, Liu E, Qian S, Liu X, Wu Y, Dong F, Qiu C-W (2021) Artificial intelligence: a powerful paradigm for scientific research. The Innovation 2(4):100179
Singh N, Sabrol H (2021) Convolutional neural networks: an extensive arena of deep learning. A comprehensive study. Arch Comput Methods Eng 28(7):4755–4780
Sarker IH (2021) Machine learning: algorithms, real-world applications and research directions. SN Comput Sci 2(3):160
Wang P, Fan E, Wang P (2021) Comparative analysis of image classification algorithms based on traditional machine learning and deep learning. Pattern Recogn Lett 141:61–67
Sun L, Chen J, Feng D, Xing M (2021) The recognition framework of deep kernel learning for enclosed remote sensing objects. IEEE Access 9:95585–95596
Ai J, Tian R, Luo Q, Jin J, Tang B (2019) Multi-scale rotation-invariant haar-like feature integrated cnn-based ship detection algorithm of multiple-target environment in sar imagery. IEEE Trans Geosci Remote Sens 57(12):10070–10087
Chen S, Li X (2019) A new CFAR algorithm based on variable window for ship target detection in sar images. SIViP 13(4):779–786
Xu Y, Zhang X, Wei S, Shi J, Zeng T, Zhang T (2023) A target-oriented Bayesian compressive sensing imaging method with region-adaptive extractor for mmw automotive radar. IEEE Trans Geosci Remote Sensing
Nasrabadi NM (2019) Deeptarget: an automatic target recognition using deep convolutional neural networks. IEEE Trans Aerosp Electron Syst 55(6):2687–2697
Guo Y, Du L, Lyu G (2021) Sar target detection based on domain adaptive faster r-cnn with small training data size. Remote Sensing 13(21):4202
Wei X, Zhang S, Qi Q, Fu H, Qiu T, Zhou A (2021) Predicting malignancy and benign thyroid nodule using multi-scale feature fusion and deep learning. Pattern Recognit Image Anal 31:830–841
Yasir M, Jianhua W, Mingming X, Hui S, Zhe Z, Shanwei L, Colak ATI, Hossain MS (2023) Ship detection based on deep learning using sar imagery: a systematic literature review. Soft Comput 27(1):63–84
Liu L, Chen G, Pan Z, Lei B, An Q (2018) Inshore ship detection in sar images based on deep neural networks. In: IGARSS 2018-2018 IEEE International Geoscience and Remote Sensing Symposium, pp. 25–28. IEEE
Lin Z, Ji K, Leng X, Kuang G (2018) Squeeze and excitation rank faster r-cnn for ship detection in sar images. IEEE Geosci Remote Sens Lett 16(5):751–755
Xu X, Zhang X, Shao Z, Shi J, Wei S, Zhang T, Zeng T (2022) A group-wise feature enhancement-and-fusion network with dual-polarization feature enrichment for sar ship detection. Remote Sensing 14(20):5276
Ma P, Li C, Rahaman MM, Yao Y, Zhang J, Zou S, Zhao X, Grzegorzek M (2023) A state-of-the-art survey of object detection techniques in microorganism image analysis: from classical methods to deep learning approaches. Artif Intell Rev 56(2):1627–1698
Gao Y, Wu Z, Ren M, Wu C (2022) Improved yolov4 based on attention mechanism for ship detection in sar images. IEEE Access 10:23785–23797
Guo Y, Chen S, Zhan R, Wang W, Zhang J (2022) Sar ship detection based on yolov5 using cbam and bifpn. In: IGARSS 2022-2022 IEEE International Geoscience and Remote Sensing Symposium, pp. 2147–2150. IEEE
Xu X, Zhang X, Zhang T (2022) Lite-yolov5: a lightweight deep learning detector for on-board ship detection in large-scene sentinel-1 SAR images. Remote Sensing 14(4):1018
Yang L, Liu Y, Yu H, Fang X, Song L, Li D, Chen Y (2021) Computer vision models in intelligent aquaculture with emphasis on fish detection and behavior analysis: a review. Arch Comput Methods Eng 28:2785–2816
Sun Z, Lei Y, Leng X, Xiong B, Ji K (2022) An improved oriented ship detection method in high-resolution sar image based on yolov5. In: 2022 Photonics & Electromagnetics Research Symposium (PIERS), pp. 647–653. IEEE
Li S, Li M, Li R, He C, Zhang L (2023) One-to-few label assignment for end-to-end dense detection. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 7350–7359
Madjidi H, Laroussi T (2023) Approximate mle based automatic bilateral censoring CFAR ship detection for complex scenes of log-normal sea clutter in sar imagery. Digit Signal Process 136:103972
Hou Q, Zhou D, Feng J (2021) Coordinate attention for efficient mobile network design. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 13713–13722
Lai Y, Ma R, Chen Y, Wan T, Jiao R, He H (2023) A pineapple target detection method in a field environment based on improved yolov7. Appl Sci 13(4):2691
Subedi S, Bist R, Yang X, Chai L (2023) Tracking floor eggs with machine vision in cage-free hen houses. Poult Sci 102(6):102637
Wang C-Y, Bochkovskiy A, Liao H-YM (2023) Yolov7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 7464–7475
Yuan B, Sun Z, Pei L, Li W, Hu Y, Mohammed A-S (2023) Airfield concrete pavement joint detection network based on dual-modal feature fusion. Autom Constr 151:104868
Lee S-H, Bae S-H (2023) Afi-gan: Improving feature interpolation of feature pyramid networks via adversarial training for object detection. Pattern Recogn 138:109365
Mishra A, Gupta P, Tewari P (2022) Global u-net with amalgamation of inception model and improved kernel variation for MRI brain image segmentation. Multimed Tools Appl 81(16):23339–23354
Liu K, Peng L, Tang S (2023) Underwater object detection using tc-yolo with attention mechanisms. Sensors 23(5):2567
Raj GD, Prabadevi B (2023) Steel strip quality assurance with yolov7-csf: a coordinate attention and siou fusion approach. IEEE Access 11:129493–129506
Cao L, Zheng X, Fang L (2023) The semantic segmentation of standing tree images based on the yolo v7 deep learning algorithm. Electronics 12(4):929
Zhu L, Lee F, Cai J, Yu H, Chen Q (2022) An improved feature pyramid network for object detection. Neurocomputing 483:127–139
Zhang R, Xie C, Deng L (2023) A fine-grained object detection model for aerial images based on yolov5 deep neural network. Chin J Electron 32(1):51–63
Tang F, Yang F, Tian X (2023) Long-distance person detection based on yolov7. Electronics 12(6):1502
Gevorgyan Z (2022) Siou loss: More powerful learning for bounding box regression. arXiv preprint arXiv:2205.12740
Liu Z, Cai Y, Wang H, Chen L, Gao H, Jia Y, Li Y (2021) Robust target recognition and tracking of self-driving cars with radar and camera information fusion under severe weather conditions. IEEE Trans Intell Transp Syst 23(7):6640–6653
Yang X, Yan J (2020) Arbitrary-oriented object detection with circular smooth label. In: Computer Vision–ECCV 2020: 16th European Conference, Glasgow, UK, August 23–28, 2020, Proceedings, Part VIII 16, pp. 677–694. Springer
Jiao Y, Zhu Q, He H, Zhao T, Wang H (2022) Rotating target detection based on lightweight network. In: PRICAI 2022: Trends in Artificial Intelligence: 19th Pacific Rim International Conference on Artificial Intelligence, PRICAI 2022, Shanghai, China, November 10–13, 2022, Proceedings, Part III, pp. 619–630. Springer
Yang X, Yan J (2022) On the arbitrary-oriented object detection: classification based approaches revisited. Int J Comput Vision 130(5):1340–1365
Yang X, Yan J (2022) On the arbitrary-oriented object detection: classification based approaches revisited. Int J Comput Vision 130(5):1340–1365
Wang K, Liu M (2022) Yolov3-mt: a yolov3 using multi-target tracking for vehicle visual detection. Appl Intell 52(2):2070–2091
Wang C-Y, Bochkovskiy A, Liao H-YM (2023) Yolov7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 7464–7475
Zhang T, Zhang X, Li J, Xu X, Wang B, Zhan X, Xu Y, Ke X, Zeng T, Su H (2021) Sar ship detection dataset (SSDD): official release and comprehensive data analysis. Remote Sensing 13(18):3690
Zhang T, Zhang X, Shao Z (2023) Saliency-guided attention-based feature pyramid network for ship detection in sar images. In: IGARSS 2023-2023 IEEE International Geoscience and Remote Sensing Symposium, pp. 4950–4953. IEEE
Gong Y, Zhang Z, Wen J, Lan G, Xiao S (2023) Small ship detection of sar images based on optimized feature pyramid and sample augmentation. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing
Hou X, Ao W, Song Q, Lai J, Wang H, Xu F (2020) FUSAR-ship: building a high-resolution SAR-AIS matchup dataset of Gaofen-3 for ship detection and recognition. SCIENCE CHINA Inf Sci 63:1–19
Zhang L, Dong H, Zou B (2019) Efficiently utilizing complex-valued Polsar image data via a multi-task deep learning framework. ISPRS J Photogramm Remote Sens 157:59–72
Acknowledgments
The authors would like to thank the anonymous reviewers for their valuable comments. This article has been supported by the National Natural Science Foundation of China (61941113) and Science and Technology on Information System Engineering Laboratory (No: 05202104).
Funding
Not applicable.
Author information
Authors and Affiliations
Contributions
HZ and ZW were involved in conceptualization; ZW helped with methodology, validation, resources, and project administration; and HZ was involved in software, formal analysis, investigation, data curation, writing—original draft preparation, writing—review and editing, and visualization. All authors have read and agreed to the published version of the manuscript. All authors read and approved the final manuscript.
Corresponding author
Ethics declarations
Conflict of interest
The authors declare that they have no conflict of interest.
Ethics approval
Not applicable.
Consent to participate
Not applicable.
Consent for publication
We would like to confirm that this work is original and has not been published elsewhere, nor is it currently under consideration for publication elsewhere. All the authors give their consent to publish the manuscript.
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
Zou, H., Wang, Z. An enhanced object detection network for ship target detection in SAR images. J Supercomput 80, 17377–17399 (2024). https://doi.org/10.1007/s11227-024-06136-3
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11227-024-06136-3