基于改进残差收缩网络的带钢缺陷检测方法.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN113838208A(43)申请公布日2021.12.24(21)申请号202111025437.7(22)申请日2021.09.02(71)申请人桂林电子科技大学地址541004广西壮族自治区桂林市七星区金鸡路1号(72)发明人蒋占四梁日强滕制胡燕林郑泽瀚(74)专利代理机构桂林文必达专利代理事务所(特殊普通合伙)45134代理人张学平(51)Int.Cl.G06T17/20(2006.01)G06K9/62(2006.01)G06N3/04(2006.01)G06N3/08(2006.01)权利要求书1页说明书6页附图2页(54)发明名称基于改进残差收缩网络的带钢缺陷检测方法(57)摘要本发明公开了一种基于改进残差收缩网络的带钢缺陷检测方法，通过增广数据集提高网络训练的准确度，构建残差收缩网络模型，再将新的激活函数LeakyReLU和Adamax优化器替换原残差收缩网络的激活函数和优化器，并且应用于带钢的缺陷检测中，训练完成后以准确度判断检测结果，如未达到预设值则重新训练至预设目标才保存网络输出结果，所述基于改进残差收缩网络的带钢缺陷检测方法有效剔除了网络训练特征学习带来的冗余信息，提升残差收缩网络的性能，解决了现有技术中带钢识别方法准确率不高的问题。CN113838208ACN113838208A权利要求书1/1页1.一种基于改进残差收缩网络的带钢缺陷检测方法，其特征在于，包括下列步骤：导入数据集，增广数据集；搭建改进的残差收缩网络模型；选择优化器；训练完成网络并验证。2.如权利要求1所述的基于改进残差收缩网络的带钢缺陷检测方法，其特征在于，增广数据集由所述数据集随机旋转和随机水平翻转组成完成扩充。3.如权利要求1所述的基于改进残差收缩网络的带钢缺陷检测方法，其特征在于，所述改进的残差收缩网络模型中使用LeakyReLU激活函数代替传统的深度卷积网络中的ReLU激活函数。4.如权利要求1所述的基于改进残差收缩网络的带钢缺陷检测方法，其特征在于，所述改进的残差收缩网络模型中的交叉熵损失函数应用softmax函数，并把特征强制控制在(0,1)的范围内。5.如权利要求1所述的基于改进残差收缩网络的带钢缺陷检测方法，其特征在于，所述改进的残差收缩网络模型使用软阈值化消除图像噪声。6.如权利要求1所述的基于改进残差收缩网络的带钢缺陷检测方法，其特征在于，使用Adamax算法代替Adam优化算法对所述改进的残差收缩网络模型进行优化。7.如权利要求1所述的基于改进残差收缩网络的带钢缺陷检测方法，其特征在于，在训练完成网络并验证的过程中，判断准确度是否可以达到预设值，如果达到则保存网络，输出训练的结果；如果未达到则重新训练。2CN113838208A说明书1/6页基于改进残差收缩网络的带钢缺陷检测方法技术领域[0001]本发明涉及机器视觉技术领域，尤其涉及一种基于改进残差收缩网络的带钢缺陷检测方法。背景技术[0002]图像识别是机器视觉的重要部分，图像作为一种信息载体，会得到更加直观的，高效的视觉信息。近年来深度学习不断的发展，产生了各种的网络类型，并且都有在工业检测上应用。传统的视觉的带钢识别方法一般都是图像处理或者分类器加特征提取，这些传统代表的方法存在受到图像质量以及图像灰度的影响，以及识别精度比较低。发明内容[0003]本发明的目的在于提供一种基于改进残差收缩网络的带钢缺陷检测方法，旨在解决现有技术中带钢识别方法准确率不高的问题。[0004]为实现上述目的，本发明采用一种基于改进残差收缩网络的带钢缺陷检测方法，包括下列步骤：[0005]导入数据集，增广数据集；[0006]搭建改进的残差收缩网络模型；[0007]选择优化器；[0008]训练完成网络并验证。[0009]其中，增广数据集由所述数据集随机旋转和随机水平翻转组成完成扩充。[0010]其中，所述改进的残差收缩网络模型中使用LeakyReLU激活函数代替传统的深度卷积网络中的ReLU激活函数。[0011]其中，所述改进的残差收缩网络模型中的交叉熵损失函数应用softmax函数，并把特征强制控制在(0,1)的范围内。[0012]其中，所述改进的残差收缩网络模型使用软阈值化消除图像噪声。[0013]其中，使用Adamax算法代替Adam优化算法对所述改进的残差收缩网络模型进行优化。[0014]其中，在训练完成网络并验证的过程中，判断准确度是否可以达到预设值，[0015]如果达到则保存网络，输出训练的结果；[0016]如果未达到则重新训练。[0017]本发明的一种基于改进残差收缩网络的带钢缺陷检测方法，通过增广数据集提高网络训练的准确度，构建残差收缩网络模型，再将新的激活函数LeakyReLU和Adama