[go: up one dir, main page]
More Web Proxy on the site http://driver.im/ Skip to main content
Log in

An enhanced object detection network for ship target detection in SAR images

  • Published:
The Journal of Supercomputing Aims and scope Submit manuscript

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.

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

Access this article

Subscribe and save

Springer+ Basic
£29.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 includes VAT (United Kingdom)

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

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

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

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

    MathSciNet  Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

  9. Sarker IH (2021) Machine learning: algorithms, real-world applications and research directions. SN Comput Sci 2(3):160

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

  15. Nasrabadi NM (2019) Deeptarget: an automatic target recognition using deep convolutional neural networks. IEEE Trans Aerosp Electron Syst 55(6):2687–2697

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

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

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

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

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

    Google Scholar 

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

    MathSciNet  Google Scholar 

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

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

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

    Google Scholar 

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

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

    Google Scholar 

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

    Google Scholar 

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

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

    Google Scholar 

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

    Google Scholar 

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

    Google Scholar 

  37. Liu K, Peng L, Tang S (2023) Underwater object detection using tc-yolo with attention mechanisms. Sensors 23(5):2567

    Google Scholar 

  38. Raj GD, Prabadevi B (2023) Steel strip quality assurance with yolov7-csf: a coordinate attention and siou fusion approach. IEEE Access 11:129493–129506

    Google Scholar 

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

    Google Scholar 

  40. Zhu L, Lee F, Cai J, Yu H, Chen Q (2022) An improved feature pyramid network for object detection. Neurocomputing 483:127–139

    Google Scholar 

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

    Google Scholar 

  42. Tang F, Yang F, Tian X (2023) Long-distance person detection based on yolov7. Electronics 12(6):1502

    Google Scholar 

  43. Gevorgyan Z (2022) Siou loss: More powerful learning for bounding box regression. arXiv preprint arXiv:2205.12740

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

    Google Scholar 

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

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

  47. Yang X, Yan J (2022) On the arbitrary-oriented object detection: classification based approaches revisited. Int J Comput Vision 130(5):1340–1365

    Google Scholar 

  48. Yang X, Yan J (2022) On the arbitrary-oriented object detection: classification based approaches revisited. Int J Comput Vision 130(5):1340–1365

    Google Scholar 

  49. Wang K, Liu M (2022) Yolov3-mt: a yolov3 using multi-target tracking for vehicle visual detection. Appl Intell 52(2):2070–2091

    Google Scholar 

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

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

    Google Scholar 

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

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

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

    Google Scholar 

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

    Google Scholar 

Download references

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

Authors

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

Correspondence to Haochen Zou.

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.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

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

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11227-024-06136-3

Keywords

Navigation