CCDS-YOLO: Multi-Category Synthetic Aperture Radar Image Object Detection Model Based on YOLOv5s

被引:6
|
作者
Huang, Min [1 ]
Liu, Zexu [1 ]
Liu, Tianen [1 ]
Wang, Jingyang [1 ,2 ]
机构
[1] Hebei Univ Sci & Technol, Shijiazhuang 050018, Peoples R China
[2] Hebei Technol Innovat Ctr Intelligent IoT, Shijiazhuang 050018, Peoples R China
关键词
target recognition; SAR; YOLOv5; multi-category; deep learning; SHIP DETECTION; SAR;
D O I
10.3390/electronics12163497
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Synthetic Aperture Radar (SAR) is an active microwave sensor that has attracted widespread attention due to its ability to observe the ground around the clock. Research on multi-scale and multi-category target detection methods holds great significance in the fields of maritime resource management and wartime reconnaissance. However, complex scenes often influence SAR object detection, and the diversity of target scales also brings challenges to research. This paper proposes a multi-category SAR image object detection model, CCDS-YOLO, based on YOLOv5s, to address these issues. Embedding the Convolutional Block Attention Module (CBAM) in the feature extraction part of the backbone network enables the model's ability to extract and fuse spatial information and channel information. The 1 x 1 convolution in the feature pyramid network and the first layer convolution of the detection head are replaced with the expanded convolution, Coordinate Conventional (CoordConv), forming a CRD-FPN module. This module more accurately perceives the spatial details of the feature map, enhancing the model's ability to handle regression tasks compared to traditional convolution. In the detector segment, a decoupled head is utilized for feature extraction, offering optimal and effective feature information for the classification and regression branches separately. The traditional Non-Maximum Suppression (NMS) is substituted with the Soft Non-Maximum Suppression (Soft-NMS), successfully reducing the model's duplicate detection rate for compact objects. Based on the experimental findings, the approach presented in this paper demonstrates excellent results in multi-category target recognition for SAR images. Empirical comparisons are conducted on the filtered MSAR dataset. Compared with YOLOv5s, the performance of CCDS-YOLO has been significantly improved. The mAP@0.5 value increases by 3.3% to 92.3%, the precision increases by 3.4%, and the mAP@0.5:0.95 increases by 6.7%. Furthermore, in comparison with other mainstream detection models, CCDS-YOLO stands out in overall performance and anti-interference ability.
引用
收藏
页数:22
相关论文
共 50 条
  • [31] Track Fastener Defect Detection Model Based on Improved YOLOv5s
    Li, Xue
    Wang, Quan
    Yang, Xinwen
    Wang, Kaiyun
    Zhang, Hongbing
    SENSORS, 2023, 23 (14)
  • [32] Yolo-inspection: defect detection method for power transmission lines based on enhanced YOLOv5s
    Lu, Lihui
    Chen, Zhencong
    Wang, Rifan
    Liu, Li
    Chi, Haoqing
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2023, 20 (05)
  • [33] Yolo-inspection: defect detection method for power transmission lines based on enhanced YOLOv5s
    Lihui Lu
    Zhencong Chen
    Rifan Wang
    Li Liu
    Haoqing Chi
    Journal of Real-Time Image Processing, 2023, 20
  • [34] Object Detection of UAV Images from Orthographic Perspective Based on Improved YOLOv5s
    Lu, Feng
    Li, Kewei
    Nie, Yunfeng
    Tao, Yejia
    Yu, Yihao
    Huang, Linbo
    Wang, Xing
    SUSTAINABILITY, 2023, 15 (19)
  • [35] Object detection algorithm for indoor switchgear components in substations based on improved YOLOv5s
    Changdong, Wu
    Rui, Liu
    INSIGHT, 2024, 66 (04) : 226 - 231
  • [36] Improved YOLOv5 Based on Multi-Strategy Integration for Multi-Category Wind Turbine Surface Defect Detection
    Lei, Mingwei
    Wang, Xingfen
    Wang, Meihua
    Cheng, Yitao
    ENERGIES, 2024, 17 (08)
  • [37] Ship target detection method for synthetic aperture radar images based on improved YOLOv5
    He Z.
    Li M.
    Gou Y.
    Yang A.
    Xi Tong Gong Cheng Yu Dian Zi Ji Shu/Systems Engineering and Electronics, 2023, 45 (12): : 3743 - 3753
  • [38] Yolo-tla: An Efficient and Lightweight Small Object Detection Model based on YOLOv5
    Ji, Chun-Lin
    Yu, Tao
    Gao, Peng
    Wang, Fei
    Yuan, Ru-Yue
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (04)
  • [39] UAV small target detection algorithm based on an improved YOLOv5s model
    Cao, Shihai
    Wang, Ting
    Li, Tao
    Mao, Zehui
    JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2023, 97
  • [40] A lightweight ship target detection model based on improved YOLOv5s algorithm
    Zheng, Yuanzhou
    Zhang, Yuanfeng
    Qian, Long
    Zhang, Xinzhu
    Diao, Shitong
    Liu, Xinyu
    Cao, Jingxin
    Huang, Haichao
    PLOS ONE, 2023, 18 (04):