[发明专利]工业检测中的凹痕缺陷样本生成方法、生成装置

权利要求书

1.一种工业检测中的凹痕缺陷样本生成方法，其特征在于，包括以下步骤：

步骤S1，创建一个与良品图片比例一致的平面；

步骤S2，添加两个图像纹理节点电池；

步骤S3，调用paython模块生成尺寸大小与良品图一致的法向贴图；

步骤S4，将良品图和所述法向贴图分别导入两个所述图像纹理节点电池中；

步骤S5，制作图像纹理节点电池流程；

步骤S6，调节摄像机、光照强度及光照角度；

步骤S7，基于以上步骤S1-S6在渲染器中制作渲染工作室；

步骤S8，根据所述渲染工作室编辑paython脚本代码，以实现凹痕缺陷样本的自动化合成；

步骤S9，运行所述paython脚本代码，批量生成带有标注信息的凹痕缺陷样本。

2.根据权利要求1所述的工业检测中的凹痕缺陷样本生成方法，其特征在于，制作图像纹理节点电池流程，具体包括：

将导入所述良品图的图像纹理节点电池的颜色节点连接平面基础色节点；

将导入所述法向贴图的图像纹理节点电池的颜色节点连接到控制凹痕深度的组合变量节点，将所述组合变量节点连接凹凸节点电池，将所述凹凸节点电池连接平面的法向节点。

3.根据权利要求1所述的工业检测中的凹痕缺陷样本生成方法，其特征在于，其中，采用以下方式生成获取所述标注信息：

采用cv2.imread函数读取所述法向贴图，并将所述法向贴图转化为灰度图；

将所述灰度图中灰度值大于预设值的像素点存入数组；

将所述数组中的像素点变为纯白色；

采用cv2.dilate函数对所述数组进行膨胀处理；

采用cv2.findContours函数对膨胀处理后的数组进行寻找轮廓处理；

采用numpy.asarray函数获取轮廓的轮廓标注点；

采用tolist函数将数组结构转换为标注点列表。

4.根据权利要求1所述的工业检测中的凹痕缺陷样本生成方法，其特征在于，调用paython模块生成尺寸大小与良品图一致的法向贴图，具体包括：

运用numpy.zeros函数构建尺寸大小与良品图一致的第一法向贴图；

运用numpy.exp函数和math.sqrt函数制作第一法向贴图的椭圆形状，运用numpy.fromfunction函数添加椭圆形状的渐变效果，以形成最终的法向贴图。

5.根据权利要求1所述的工业检测中的凹痕缺陷样本生成方法，其特征在于，所述渲染器为blender。

6.一种工业检测中的凹痕缺陷样本生成装置，其特征在于，包括：

制作模块，所述制作模块用于基于以下步骤S1-S6在渲染器中制作渲染工作室：

步骤S1，创建一个与良品图片比例一致的平面；

步骤S2，添加两个图像纹理节点电池；

步骤S3，调用paython模块生成尺寸大小与良品图一致的法向贴图；

步骤S4，将良品图和所述法向贴图分别导入两个所述图像纹理节点电池中；

步骤S5，制作图像纹理节点电池流程；

步骤S6，调节摄像机和光照强度及光照角度；

合成模块，所述合成模块用于根据所述渲染工作室编辑paython脚本代码，以实现凹痕缺陷样本的自动化合成；

生成模块，所述生成模块用于运行所述paython脚本代码，批量生成带有标注信息的凹痕缺陷样本。

7.根据权利要求6所述的工业检测中的凹痕缺陷样本生成装置，其特征在于，所述制作模块具体用于：

将导入所述良品图的图像纹理节点电池的颜色节点连接平面基础色节点；

将导入所述法向贴图的图像纹理节点电池的颜色节点连接到控制凹痕深度的组合变量节点，将所述组合变量节点连接凹凸节点电池，将所述凹凸节点电池连接平面的法向节点。