技术特征：
1.一种基于改进yolov5算法的目标检测方法，建立基于yolov5算法构建改进型yolov5神经网络结构并训练优化模型，用于丝饼缺陷的实时检测，其特征在于，所述该方法包括：步骤1、获取包含缺陷的丝饼检测图像数据，并进行数据预处理得到缺陷图像数据样本集；步骤2、基于yolov5算法构建改进型yolov5神经网络结构，得到改进yolov5算法；步骤3、利用所述样本集数据对所述改进yolov5算法进行训练，将训练后最佳的权重参数加载至改进yolov5算法得到优化的丝饼缺陷目标检测模型；步骤4、现场采集待检测图像并进行图像预处理，然后输入优化的丝饼缺陷目标检测模型，自动输出对应的缺陷目标检测结果；所述目标检测结果包括待检测图像中缺陷目标区域的位置以及每个目标区域对应的类别，其中目标区域的缺陷类别为绊丝、油污、成型不良。2.如权利要求1所述的基于改进yolov5算法的目标检测方法，其特征在于，所述改进型yolov5网络结构包括特征提取骨干网络backbone模块、用于预测分类和目标边界框head模块；所述特征提取骨干网络backbone模块输出3个不同尺度的特征图；其中所述backbone模块包括focus子模块、conv子模块、c3se子模块、spp子模块，所述head模块包括bifpn子模块、detect子模块；所述focus模块对输入的图像进行切片操作，用于特征提取；所述c3se结构是在通道维度引入注意力机制，用于提取不同尺度的特征图；所述spp模块用于融合多尺度的特征；所述bifpn模块，在fpn模块基础上进一步上采样c3se的输出特征进行融合，用于加强网络特征融合能力；所述detect子模块，包括3个分类器，用于将融合特征图输入多分类器进行丝饼缺陷定位及分类，给出缺陷目标预测框。3.如权利要求1所述的基于改进yolov5算法的目标检测方法，其特征在于，所述步骤1的数据预处理包括：将大小为640
×
640的4张丝饼缺陷的图片采用mosaic数据增强方法进行随机缩放、随机裁剪、随机排布，用于丰富缺陷样本数据集；所述丝饼缺陷包括绊丝、油污、成型不良的图片。4.如权利要求1所述的基于改进yolov5算法的目标检测方法，其特征在于，所述对图片进行切片操作，是将w、h信息进行分割，用于扩充通道空间，将原先的rgb三通道模式变成了12个通道，然后将得到的新图片再经过3
×
3卷积操作提取特征；w、h分别为宽、高。5.如权利要求1所述的基于改进yolov5算法的目标检测方法，其特征在于，所述c3模块包括两个分支：y1和y2，其中分支y1由依次连接的conv模块、x个添加shortcut分支的bottleneck模块、conv组成，分支y2由conv组成，最后将2个分支进行concat操作，并依次经过bn层、leakyrelu激活函数、conv模块；所述conv模块由conv、bn、silu激活函数三者组成。6.如权利要求1所述的基于改进yolov5算法的目标检测方法，其特征在于，所述c3se模块包括c3模块并在通道维度引入注意力机制se模块，通过在feature map层上执行全局平均池化，把原本h
×
w
×
c的特征，压缩为1
×1×
c，得到当前feature map的全局压缩特征量，每个通道用一个数值表示；通过两个全连接层的bottleneck结构去建模通道间的相关性，得到feature map中每个通道的权值，并将加权后的feature map作为下一层网络的输入；
将上一步得到的归一化权重加权到每个通道的特征上，与原始特征图相乘，得到最终特征图输出，完成在通道维度上引入注意力机制。7.如权利要求1所述的基于改进yolov5算法的目标检测方法，其特征在于，所述spp模块采用5
×
5、9
×
9、13
×
13的最大池化尺寸进行多尺度融合，对输入的特征图分别利用三个最大池化层进行处理，并将不同尺度的特征图与输入进行concat操作，得到池化特征图输出。8.如权利要求1所述的基于改进yolov5算法的目标检测方法，其特征在于，所述head结构中，在完成特征的初步提取后，增加上采样分支进一步对c3se输出的特征图进行融合，使特征更加具有高语义信息、增强网络特征融合能力，得到预测的特征图。9.如权利要求1所述的基于改进yolov5算法的目标检测方法，其特征在于，所述detect模块的多个分类器用于接收所述特征融合模块输出的大小为20
×
20的融合特征、大小为40
×
40的融合特征、大小为80
×
80的融合特征，在特征图上应用初始锚框输出预测框，与真实框进行比对，并生成带有类概率、对象置信度得分和目标缺陷预测框的最终输出向量；是采用非极大值抑制nms操作对多目标框进行筛选确定最终的目标缺陷预测框。10.如权利要求1所述的基于改进yolov5算法的目标检测方法，其特征在于，所述缺陷图像数据样本集中的图像数据要人工预先进行缺陷标注，再按比例分为训练集和验证集，训练集用于训练网络，采用ciou_loss计算bounding box损失函数并与预设阈值比较，从而回调网络参数；所述验证集用于验证模型的泛化能力，得到最佳的权重参数。
技术总结
本发明涉及一种基于YOLOv5神经网络的丝饼缺陷检测方法。包括：获取丝饼图像缺陷数据集，并对丝饼图像缺陷数据集进行数据处理得到样本集；基于YOLOv5算法构建改进型YOLOv5神经网络结构，得到改进YOLOv5算法；利用样本集对改进YOLOv5算法进行训练，将训练后最佳的权重文件加载至改进YOLOv5算法得到目标检测模型；针对获取的待分类图像，采用所述目标检测模型输出对应的丝饼缺陷定位及分类。本发明的方法相较于原YOLOv5算法具有较高的检测速度和小目标检测精度，从而有效提高生产效率，降低了丝饼缺陷检测的错误率。丝饼缺陷检测的错误率。丝饼缺陷检测的错误率。
技术研发人员：孟祥茹 孙咏 王嵩 于碧辉 王美吉
受保护的技术使用者：中国科学院沈阳计算技术研究所有限公司
技术研发日：2021.08.11
技术公布日：2023/2/17