表面划痕检测神经网络的构建方法、系统、装置及介质制造方法及图纸

本发明专利技术公开了一种表面划痕检测神经网络的构建方法、系统、装置及介质。该方法通过获取物体表面带有划痕的图像数据，基于预设的分辨率阈值对所述图像数据进行筛选，得到划痕图像样本；然后通过特征提取模块、分割网络模块和决策网络模块构建基础神经网络；最终通过所述划痕图像样本训练所述基础神经网络，得到表面划痕检测神经网络。该系统包括获取单元、筛选单元、构建单元和训练单元。通过使用本发明专利技术中的方法，识别出有划痕缺陷的物体图像，并且在网络复杂度低的同时实现了少量样本训练与高精度的检测。本发明专利技术可广泛应用于图像识别技术领域内。

Construction method, system, device and medium of neural network for surface scratch detection

【技术实现步骤摘要】
表面划痕检测神经网络的构建方法、系统、装置及介质
本专利技术涉及图像识别
，尤其是一种表面划痕检测神经网络的构建方法、系统、装置及介质。
技术介绍
随着深度学习领域的快速发展，自动表面缺陷检测技术成为了工业应用上的重要检测技术。但是由于一般的工业实际生产中只能提供较少的缺陷样本，因此如何利用有限的样本得到所需要的特征进行训练，构建出合适的模型达到较高的精度是一个重要问题。而现有的基于深度学习的物体表面缺陷检测技术，在追求检测精度的过程中，往往存在计算量大和训练困难等问题。如LBP特征提取方法，虽然在数据量较少的情况下能较好地提取图像的信息，但是算法计算复杂量大，在计算对应关系的时候难以选取合适的稀疏字典。而基于卷积神经网络的方法中，则通常是在浅层网络的基础上加深网络深度，以更复杂的网络来提高检测的精度，但这却降低了网络的训练效率，检测费时很长。目前，现有方案中还缺少一种良好的神经网络构建技术，以用于解决以上问题。
技术实现思路
本专利技术的目的在于至少一定程度上解决现有技术中存在的技术问题之一。为此，本本文档来自技高网...

【技术保护点】
1.表面划痕检测神经网络的构建方法，其特征在于，包括以下步骤：/n获取物体表面带有划痕的图像数据，基于预设的分辨率阈值对所述图像数据进行筛选，得到划痕图像样本；/n通过特征提取模块、分割网络模块和决策网络模块构建基础神经网络；/n通过所述划痕图像样本训练所述基础神经网络，得到表面划痕检测神经网络；/n其中，所述特征提取模块用于提取所述划痕图像样本的高频信息，并将结果输入到分割网络模块；所述分割网络模块用于对所述划痕图像样本进行像素级别的语义分割，并将结果输入到决策网络模块；所述决策网络模块用于输出特征向量。/n

【技术特征摘要】
1.表面划痕检测神经网络的构建方法，其特征在于，包括以下步骤：
获取物体表面带有划痕的图像数据，基于预设的分辨率阈值对所述图像数据进行筛选，得到划痕图像样本；
通过特征提取模块、分割网络模块和决策网络模块构建基础神经网络；
通过所述划痕图像样本训练所述基础神经网络，得到表面划痕检测神经网络；
其中，所述特征提取模块用于提取所述划痕图像样本的高频信息，并将结果输入到分割网络模块；所述分割网络模块用于对所述划痕图像样本进行像素级别的语义分割，并将结果输入到决策网络模块；所述决策网络模块用于输出特征向量。


2.根据权利要求1所述的表面划痕检测神经网络的构建方法，其特征在于，所述特征提取模块包括粗糙提取子模块和局部残差子模块，所述粗糙提取子模块提取的粗糙图像特征信息用于输入到局部残差子模块。


3.根据权利要求2所述的表面划痕检测神经网络的构建方法，其特征在于，所述粗糙提取子模块包括两个卷积核大小为3×3的卷积层。


4.根据权利要求2所述的表面划痕检测神经网络的构建方法，其特征在于，所述特征提取模块还包括注意力子模块；
所述局部残差子模块对接收到的所述粗糙图像特征信息进行处理，得到残差图像特征信息，并将所述残差图像特征信息传递给对应的注意力子模块；
将所述残差图像特征信息在注意力子模块中与注意力掩模点乘操作，提取得到关注区域特征信息；
将所述关注区域特征信息和残差图像特征信息进行融合，并输入到所述分割网络模块。


5.根据权利要求2-4中任一项所述的表面划痕检测神经网络的构建方法，其特征在于：
在所述局部残差子模块设置至少两个卷积层，并剔除批量标准化层；
在所述局部残差子模块中...

【专利技术属性】
技术研发人员：王永斌，刘廉如，张忠平，丁雷，
申请(专利权)人：宜通世纪物联网研究院广州有限公司，
类型：发明
国别省市：广东;44