一种改进U2-Net的瓷砖表面缺陷分割方法技术

本发明专利技术涉及一种改进U2‑

【技术实现步骤摘要】
一种改进U2‑
Net的瓷砖表面缺陷分割方法


[0001]本专利技术涉及一种改进U2‑
Net的瓷砖表面缺陷分割方法，属于显著性目标分割


技术介绍

[0002]显著性目标分割又叫显著性目标检测，主要目标为区分图像中最明显的区域。通过提取出图像中的目标区域来进行分析。目前在场景物体分割、人体前景分割、人脸人体parsing、三维重建技术等在智能安防、无人驾驶、安防监控等领域应用广泛。
[0003]传统的显著性目标检测算法整体上主要都是基于低级视觉特征的，包括中心偏差、对比度先验和背景先验等。Achanata等人分别对亮度和颜色两种低级特征处理，使用高斯差分函数来进行频域滤波，从而来计算当前像素与其周围不同大小的域中像素的对比度，以此确定图像像素的显著值。Klein等人是利用信息论中的K
‑
L散度去衡量图像的特征空间的中心位置于其周围特征的差异。这些算法都是从考虑局部像素对比的方面，它在检测目标的边缘信息时能有很好的表现，但是对目标整体很难实现检测。而基于全局对比的检测算法则是计算像素区域相对于图像里所有像素的对比，这样能够将显著目标的具体位置检测出来。如Wei等人将背景连接度的概念提出，在他们的方法中用来获取有效范围并且计算图像显著性的就是全局对比度。在高等人的方法中，先获得超像素分割后的图像，然后在CIELab颜色空间中计算纹理细节和图像的全局对比度，借助了多核增强学习融合这两个特征得到显著图，并且用使用滤波器的后处理方法优化得到的预测图。Perazzi等人首先将图像分解成不同的块，这些块与块之间的图片关系是紧凑的，具有视觉平滑性，然后给这些块的唯一特殊性和空间分布情况计算评估，根据评估度量结果生成图像的显著性图。此外还有研究人员通过构建图模型来进行图像像素的显著值计算。Yang等人据此提出了基于流型排序的SOD算法。他们的做法是首先采用超像素分割图片，然后选取合适的背景种子点，其他节点和种子点之间可以计算相似度，不同大小的排序的到初始显著图，再将它作为前景种子点，再次计算相似度再排序得到结果显著图。但是这种算法最后是否能很好地检测出显著图和种子点的选取十分相关。Yuan等人提出了一个显著性反转校正过程，去除边界附近的前景超像素，防止显著反转，从而提高基于边界先验的显著性预测的准确性和鲁棒性。Shan等人利用背景权重图为流形排序提供种子，并且利用一个三阶光滑性框架来提高流形排序的性能。在Wu等人针对先前存在的方法在检测一些背景对比度低的图片时可能遗漏目标区域的问题，提出了一种新的传播模型，该模型考虑到了利用变形平滑约束。该模型局部正则化图像的节点和它周围的点，然后使用变形的平滑约束阻止可能出现错误的结果传播下去。
[0004]总之，基于传统的目标检测算法大多提取的特征比较简单，并且对图片显著值的训练过程也不是很复杂，目前在许多场景下也有着比较好的性能。然而在大量的数据集和实际场景图像中，需要检测的环境和图片情况变得越来越复杂，检测性能和检测速率的要求在增高，基于传统的方法逐渐难以满足这些要求。

技术实现思路

[0005]为了克服现有技术中存在的缺陷，本专利技术旨在提供一种改进U2‑
Net的瓷砖表面缺陷分割方法。
[0006]本专利技术解决上述技术问题所提供的技术方案是：一种改进U2‑
Net的瓷砖表面缺陷分割方法，包括以下步骤：
[0007]步骤S1、获取瓷砖表面缺陷检测数据集，并将其分为训练集与测试集；
[0008]步骤S2、构建基于U2‑
Net的瓷砖表面缺陷分割网络模型；
[0009]所述瓷砖表面缺陷分割网络模型为六层U型结构，包括6级编码器、5级解码器、2级多尺度特征融合模块及显著图融合模块；
[0010]其中前4个编码器及对应的4个解码器由特征提取结构DCRSU构成，每个DCRSU的层数随着encoder和decoder的层数的增加而减少，即前4个编码器使用的分别是DCRSU
‑
7、DCRSU
‑
6、DCRSU
‑
5、DCRSU
‑
4，前4个解码器同理如此；
[0011]第5个编码器及对应的解码器采用的就是RSU
‑
4F；
[0012]第6个编码器引入SKnet作为最深一层的编码器；
[0013]分别在第5个解码器和第4个解码器的输入引入2个改进注意力门模块，将第6个编码器与第5个编码器输出的特征进行融合输入第5个解码器，将第4个编码器与第5个解码器输出的特征进行融合输入第4个解码器；然后使用3
×
3卷积和sigmoid函数从第6个编码器、第5个解码器、第4个解码器、第3个解码器、第2个解码器和第1个解码器生成6个输出显著概率图；然后将输出的显著图的逻辑图向上采样至于输入图像大小一致，并通过级联操作相融合；最后通过1
×
1卷积层和sigmoid函数，以生成最终的显著性概率映射图；
[0014]步骤S3、通过训练集对瓷砖表面缺陷分割网络模型进行不断迭代训练，直到网络最终收敛，得到训练好的瓷砖表面缺陷分割网络模型；
[0015]步骤S4、将待处理的图片输入到训练好的瓷砖表面缺陷分割网络模型中，得到分割的目标。
[0016]进一步的技术方案是，所述步骤S10中训练集与测试集按4：1进行划分。
[0017]进一步的技术方案是，所述DCRSU由输入卷积层、编码器、解码器和残差结构4部分组成；
[0018]输入卷积层用于提取局部特征和转换通道；
[0019]编码阶段，最后一个编码器采用卷积+批量归一化+ReLU激活函数结构，倒数第2层采用深度可分离卷积+批量归一化+ReLU激活函数结构；其余编码器利用残差结构将深度可分离卷积提取的特征和经注意力机制模块处理的输入特征相加后再输入下一个特征提取层进行特征提取,使得每级输出特征可以在聚焦于具有更多有效特征信息通道，加强每一级有效特征的提取能力并获取多尺度特征信息；
[0020]残差结构将输入层和中间层进行融合，对两个不同尺度的特征进行一个拼接；
[0021]解码阶段，解码器模块将经过拼接的特征图，经过一个3
×
3卷积一个批量归一化层和Relu激活函数通过上采样逐步修复分割对象的细节和空间维度；经最后一个解码器输出的特征图，与经输入卷积层的特征图相加融合得到经DCRSU模块处理后的最终特征图。
[0022]进一步的技术方案是，所述RSU
‑
4F将下采样和上采样换成了膨胀卷积，输入特征CxHxW首先通过2个由卷积+批量归一化层+Relu组成的模块，然后经过膨胀卷积依次为1、2、
4、8，整个过程中特征图大小不变。
[0023]进一步的技术方案是，SKnet作为最深一层的编码器，提取多尺度特征操作是将原特征图分别通过一个3
×
3的分组/深度卷积和3
×
3的空洞卷积生成两个特征图：本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种改进U2‑
Net的瓷砖表面缺陷分割方法，其特征在于，包括以下步骤：步骤S1、获取瓷砖表面缺陷检测数据集，并将其分为训练集与测试集；步骤S2、构建基于U2‑
Net的瓷砖表面缺陷分割网络模型；所述瓷砖表面缺陷分割网络模型为六层U型结构，包括6级编码器、5级解码器、2级多尺度特征融合模块及显著图融合模块；其中前4个编码器及对应的4个解码器由特征提取结构DCRSU构成，每个DCRSU的层数随着encoder和decoder的层数的增加而减少，即前4个编码器使用的分别是DCRSU
‑
7、DCRSU
‑
6、DCRSU
‑
5、DCRSU
‑
4，前4个解码器同理如此；第5个编码器及对应的解码器采用的就是RSU
‑
4F；第6个编码器引入SKnet作为最深一层的编码器；分别在第5个解码器和第4个解码器的输入引入2个改进注意力门模块，将第6个编码器与第5个编码器输出的特征进行融合输入第5个解码器，将第4个编码器与第5个解码器输出的特征进行融合输入第4个解码器；然后使用3
×
3卷积和sigmoid函数从第6个编码器、第5个解码器、第4个解码器、第3个解码器、第2个解码器和第1个解码器生成6个输出显著概率图；然后将输出的显著图的逻辑图向上采样至于输入图像大小一致，并通过级联操作相融合；最后通过1
×
1卷积层和sigmoid函数，以生成最终的显著性概率映射图；步骤S3、通过训练集对瓷砖表面缺陷分割网络模型进行不断迭代训练，直到网络最终收敛，得到训练好的瓷砖表面缺陷分割网络模型；步骤S4、将待处理的图片输入到训练好的瓷砖表面缺陷分割网络模型中，得到分割的目标。2.根据权利要求1所述的一种改进U2‑
Net的瓷砖表面缺陷分割方法，其特征在于，所述步骤S10中训练集与测试集按4：1进行划分。3.根据权利要求1所述的一种改进U2‑
Net的瓷砖表面缺陷分割方法，其特征在于，所述DCRSU由输入卷积层、编码器、解码器和残差结构4部分组成；输入卷积层用于提取局部特征和转换通道；编码阶段，最后一个编码器采用卷积+批量归一化+ReLU激活函数结构，倒数第2层采用深度可分离卷积+批量归一化+ReLU激活函数结构；其余编码器利用残差结构将深度可分离卷积提取的特征和经注意力机制模块处理的输入特征相加后再输入下一个特征提取层进行特征提取,使得每级输出特征可以在聚焦于具有更多有效特征信息通道，加强每一级有效特征的提取能力并获取多尺度特征信息；残差结构将输入层和中间层进行融合，对两个不同尺度的特征进行一个拼接；解码阶段，解码器模块将经过拼接的特征图，经...

【专利技术属性】
技术研发人员：黄进，包锐，王逢港，谢艺玮，方铮，冯义从，李剑波，荣鹏，郭伦，翟树红，
申请(专利权)人：西南交通大学，
类型：发明
国别省市：