一种基于改进YOLOv5的通信信号检测与识别方法

1.一种基于改进yolov5的通信信号检测与识别方法，其特征在于，该方法包括如下步骤：

2.根据权利要求1所述的基于改进yolov5的通信信号检测与识别方法，其特征在于，该方法具体包括以下4部分，

3.根据权利要求2所述的基于改进yolov5的通信信号检测与识别方法，其特征在于，所述在yolov5的主干网络中使用轻量化结构，该轻量化结构对yolov5中的标准conv单元、cbl单元、res unit单元、csp单元作出了轻量化改进，具体包括：

4.根据权利要求2所述的基于改进yolov5的通信信号检测与识别方法，其特征在于，所述针对性尺度特征融合模块，使用基于bifpn的针对性尺度特征融合模块；基于bifpn的针对性尺度特征融合模块隔断了小尺度特征向大尺度特征融合的路径，大尺度特征仅与适中尺度特征融合；针对性尺度特征融合模块以适中尺度为主，采用bifpn中的双向结构从适中尺度向上与向下分别融合特征，实现大尺度、小尺度特征提取；同时在适中尺度上增加一条额外路径，使三种尺度特征再次融合，从中提取最终的适中尺度特征。

5.根据权利要求2所述的基于改进yolov5的通信信号检测与识别方法，其特征在于，所述使用基于simam注意力机制构成的特征注意模块；基于simam注意力机制的特征注意模块通过可分性搭配激活函数sigmoid实现对每一个神经元的权重分配，完成空间与通道注意力分配，添加在yolov5中不同尺度的特征的融合处以及基于bifpn的针对性尺度特征融合模块中对不同尺度的特征提取处。

6.根据权利要求1所述的基于改进yolov5的通信信号检测与识别方法，其特征在于，所述对通信信号时频图进行预处理包括：对数据集进行加噪声、改变亮度、裁剪、平移、旋转和缩放处理；对时频图显示的时长范围、带宽范围、采样频率、功率显示范围调整。

7.根据权利要求1所述的基于改进yolov5的通信信号检测与识别方法，其特征在于，所述根据通信信号的时长、带宽、载频、存在性、调制方式的参数进行标注包括：采用图形标注工具对训练集中的各个图像进行标注，采用代码按照规则自动生成标注文本文件。