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 条
  • [41] Object oriented method for detection of inundation extent using multi-polarized synthetic aperture radar image
    Shen, Guozhuang
    Guo, Huadong
    Liao, Jingjuan
    JOURNAL OF APPLIED REMOTE SENSING, 2008, 2
  • [42] Detection Method of Marine Biological Objects Based on Image Enhancement and Improved YOLOv5S
    Li, Peng
    Fan, Yibing
    Cai, Zhengyang
    Lyu, Zhiyu
    Ren, Weijie
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2022, 10 (10)
  • [43] Object oriented method for detection of inundation extent using multi-polarized synthetic aperture radar image
    Shen, Guozhuang
    Guo, Huadong
    Liao, Jingjuan
    Journal of Applied Remote Sensing, 2008, 2 (01):
  • [44] A Kitchen Standard Dress Detection Method Based on the YOLOv5s Embedded Model
    Zhou, Ziyun
    Zhou, Chengjiang
    Pan, Anning
    Zhang, Fuqing
    Dong, Chaoqun
    Liu, Xuedong
    Zhai, Xiangshuai
    Wang, Haitao
    APPLIED SCIENCES-BASEL, 2023, 13 (04):
  • [45] Surface defect detection of wind turbine based on lightweight YOLOv5s model
    Zhang, Yinsheng
    Yang, Yulong
    Sun, Jiaqi
    Ji, Ru
    Zhang, Peiyan
    Shan, Huilin
    MEASUREMENT, 2023, 220
  • [46] Lightweight detection model for coal gangue identification based on improved YOLOv5s
    Shang, Deyong
    Lv, Zhibin
    Gao, Zehua
    Li, Yuntao
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (04)
  • [47] Detection of Broken Hongshan Buckwheat Seeds Based on Improved YOLOv5s Model
    Li, Xin
    Niu, Wendong
    Yan, Yinxing
    Ma, Shixing
    Huang, Jianxun
    Wang, Yingmei
    Chang, Renjie
    Song, Haiyan
    AGRONOMY-BASEL, 2024, 14 (01):
  • [48] Real-time detection model of highway vehicle based on YOLOv5s
    Liu, Yuan-Feng
    Ji, Hai-Jun
    Liu, Li-Bo
    CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2022, 37 (09) : 1228 - 1241
  • [49] Improved algorithm for pedestrian detection of lane line based on YOLOv5s model
    Shen, Guoxin
    Li, Xuerong
    Weil, Yi
    2022 IEEE 6TH ADVANCED INFORMATION TECHNOLOGY, ELECTRONIC AND AUTOMATION CONTROL CONFERENCE (IAEAC), 2022, : 405 - 409
  • [50] A Lightweight Object Detection Algorithm for Remote Sensing Images Based on Attention Mechanism and YOLOv5s
    Liu, Pengfei
    Wang, Qing
    Zhang, Huan
    Mi, Jing
    Liu, Youchen
    REMOTE SENSING, 2023, 15 (09)