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 条
  • [11] SHIP-YOLO: A Lightweight Synthetic Aperture Radar Ship Detection Model Based on YOLOv8n Algorithm
    Luo, Yonghang
    Li, Ming
    Wen, Guihao
    Tan, Yunfei
    Shi, Chaoshan
    IEEE ACCESS, 2024, 12 : 37030 - 37041
  • [12] YOLOV5s object detection based on Sim SPPF hybrid pooling
    Dong, Xiuhuan
    Li, Shixin
    Zhang, Jixiang
    OPTOELECTRONICS LETTERS, 2024, 20 (06) : 367 - 371
  • [13] Research on Winter Jujube Object Detection Based on Optimized Yolov5s
    Feng, Junzhe
    Yu, Chenhao
    Shi, Xiaoyi
    Zheng, Zhouzhou
    Yang, Liangliang
    Hu, Yaohua
    AGRONOMY-BASEL, 2023, 13 (03):
  • [14] Object Detection of Overhead Transmission Lines Based on Improved YOLOv5s
    Gu, Juping
    Hu, Junjie
    Jiang, Ling
    Wang, Zixu
    Zhang, Xinsong
    Xu, Yiming
    Zhu, Jianhong
    2022 12TH INTERNATIONAL CONFERENCE ON POWER AND ENERGY SYSTEMS, ICPES, 2022, : 388 - 392
  • [15] YOLOV5s object detection based on Sim SPPF hybrid pooling
    DONG Xiuhuan
    LI Shixin
    ZHANG Jixiang
    Optoelectronics Letters, 2024, 20 (06) : 367 - 371
  • [16] Remote Sensing Object Detection Methods Based on Improved YOLOv5s
    Cheng Kailun
    Hu Xiaobing
    Chen Haijun
    Li Hu
    LASER & OPTOELECTRONICS PROGRESS, 2024, 61 (18)
  • [17] MDP-YOLO: A LIGHTWEIGHT YOLOV5S ALGORITHM FOR MULTI-SCALE PEST DETECTION
    Yu, Jianghua
    Zhang, Bing
    ENGENHARIA AGRICOLA, 2023, 43 (04):
  • [18] YOLOv5s-DSD: An Improved Aerial Image Detection Algorithm Based on YOLOv5s
    Sun, Chaoyue
    Chen, Yajun
    Xiao, Ci
    You, Longxiang
    Li, Rongzhen
    SENSORS, 2023, 23 (15)
  • [19] CaSnLi-YOLO: construction site multi-target detection method based on improved YOLOv5s
    Hao, Fei
    Zhang, Tianqi
    He, Guanghong
    Dou, Ruofei
    Meng, Chao
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2024, 35 (08)
  • [20] YOlOv5s-ACE: Forest Fire Object Detection Algorithm Based on Improved YOLOv5s
    Wang, Jianan
    Wang, Changzhong
    Ding, Weiping
    Li, Cheng
    FIRE TECHNOLOGY, 2024, : 4023 - 4043