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 条
  • [1] CRAS-YOLO: A Novel Multi-Category Vessel Detection and Classification Model Based on YOLOv5s Algorithm
    Zhao, Wenxiao
    Syafrudin, Muhammad
    Fitriyani, Norma Latif
    IEEE ACCESS, 2023, 11 : 11463 - 11478
  • [2] LTEA-YOLO: An Improved YOLOv5s Model for Small Object Detection
    Li, Bo
    Huang, Shengbao
    Zhong, Guangjin
    IEEE ACCESS, 2024, 12 : 99768 - 99778
  • [3] MF-YOLO: Multimodal Fusion for Remote Sensing Object Detection Based on YOLOv5s
    Li, Wenqiang
    Li, Aimin
    Kong, Xiaotong
    Zhang, Yuechen
    Li, Zhiyao
    PROCEEDINGS OF THE 2024 27 TH INTERNATIONAL CONFERENCE ON COMPUTER SUPPORTED COOPERATIVE WORK IN DESIGN, CSCWD 2024, 2024, : 897 - +
  • [4] Lightweight aerial image object detection algorithm based on improved YOLOv5s
    Deng, Lixia
    Bi, Lingyun
    Li, Hongquan
    Chen, Haonan
    Duan, Xuehu
    Lou, Haitong
    Zhang, Hongyu
    Bi, Jingxue
    Liu, Haiying
    SCIENTIFIC REPORTS, 2023, 13 (01)
  • [5] Lightweight aerial image object detection algorithm based on improved YOLOv5s
    Lixia Deng
    Lingyun Bi
    Hongquan Li
    Haonan Chen
    Xuehu Duan
    Haitong Lou
    Hongyu Zhang
    Jingxue Bi
    Haiying Liu
    Scientific Reports, 13
  • [6] MDD-YOLOv8: A Multi-Scale Object Detection Model Based on YOLOv8 for Synthetic Aperture Radar Images
    Liu, Jie
    Liu, Xue
    Chen, Huaixin
    Luo, Sijie
    APPLIED SCIENCES-BASEL, 2025, 15 (04):
  • [7] YOLOv5s-Fog: An Improved Model Based on YOLOv5s for Object Detection in Foggy Weather Scenarios
    Meng, Xianglin
    Liu, Yi
    Fan, Lili
    Fan, Jingjing
    SENSORS, 2023, 23 (11)
  • [8] EMO-YOLO: a lightweight ship detection model for SAR images based on YOLOv5s
    Pan, Hao
    Guan, Shaopeng
    Jia, Wanhai
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (8-9) : 5609 - 5617
  • [9] Road object detection algorithm based on improved YOLOv5s
    Zhou Qing
    Tan Gong-quan
    Yin Song-lin
    Li Yi-nian
    Wei Dan-qin
    CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2023, 38 (05) : 680 - 690
  • [10] EST-YOLOv5s: SAR Image Aircraft Target Detection Model Based on Improved YOLOv5s
    Huang, Min
    Yan, Weihao
    Dai, Wenhui
    Wang, Jingyang
    IEEE ACCESS, 2023, 11 : 113027 - 113041