反光擦除网络的训练及其反光擦除方法、装置和设备与流程

本技术涉及深度学习，尤其涉及一种反光擦除网络的训练及其反光擦除方法、装置和设备。

背景技术：

1、在工业制造中，高光泽度材料常用于加工一些具有一定平整度和表面纹理清晰的工件。这类大多数工件通常表面较为光滑，在进行表面检测时通常需要使用光源对工件进行补光处理。然而有一些工件表面不仅光滑，还具有高反光等特性，这就使得获取到的图片表面反光严重，导致工件的缺陷表面不清晰，无法在图片中获取到工件反光部位的细节。从而影响到检测时会影响被测物的缺陷特征提取，使得缺陷检测结果出现错检漏检，降低这类工件表面缺陷检测成功率，对工件的缺陷检测造成了很大的困难。

2、因此，如何对这类工件进行反光去除，以准确进行表面缺陷检测，仍然需要提供进一步的解决方案。

技术实现思路

1、本技术的多个方面提供一种反光擦除网络的训练及其反光擦除方法、装置和设备，用于解决对具备反光特性的工件的缺陷检测成功率较低的问题。

2、本技术实施例提供一种反光擦除网络的训练方法，包括：获取多个反光工件在多个不同成像效果下的原始反光图像，以及所述多个反光工件的原始无反光图像；建立多个反光工件中各反光工件的原始反光图像和原始无反光图像之间的映射关系，以获取反光擦除网络训练集；基于所述反光擦除网络训练集，训练得到反光擦除网络，所述反光擦除网络包括第一生成器、第二生成器、第一判别器和第二判别器，所述第一生成器与所述第一判别器连接，所述第二生成器和所述第二判别器连接，多个所述反光工件中各反光工件的原始反光图像作为所述第一生成器的输入，多个所述反光工件中各反光工件的原始无反光图像作为所述第二生成器的输入；其中，所述第一生成器的网络模块和所述第二生成器的网络模块的网络结构相同，包括编码器、辅助分类器和解码器；所述第一判别器的网络模块和所述第二判别器的网络模块的网络结构相同，包括编码器和辅助分类器；所述第一生成器用于基于输入的反光工件的原始反光图像生成所述输入的反光工件的无反光图像，所述第一判别器用于确定所述第一生成器模块生成的所述输入的反光工件的无反光图像、与所述输入的反光工件的原始无反光图像之间的相似度；所述第二生成器用于基于输入的反光工件的原始无反光图像生成所述输入的反光工件的反光图像，所述第二判别器用于确定所述第二生成器模块生成的所述输入的反光工件的反光图像、与所述输入的反光工件的原始反光图像之间的相似度。

3、本技术实施例还提供一种反光擦除方法，包括：获取待反光擦除的目标工件图像；将所述目标工件图像输入至反光检测网络中，以输出所述目标工件图像中的反光检测结果，所述反光检测网络为基于多个反光图像和对应的反光区域标注训练得到的；基于反光擦除网络和所述目标工件图像的反光检测结果，对所述目标工件图像进行反光擦除操作，得到反光擦除后的目标工件图像，所述反光擦除网络为基于反光擦除网络的训练方法训练得到的。

4、本技术实施例还提供一种反光擦除网络的训练装置，包括：图像获取模块，用于获取多个反光工件在多个不同成像效果下的原始反光图像，以及所述多个反光工件的原始无反光图像；关系建立模块，用于建立多个反光工件中各反光工件的原始反光图像和原始无反光图像之间的映射关系，以获取反光擦除网络训练集；网络训练模块，用于基于所述反光擦除网络训练集，训练得到反光擦除网络，所述反光擦除网络包括第一生成器、第二生成器、第一判别器和第二判别器，所述第一生成器与所述第一判别器连接，所述第二生成器和所述第二判别器连接，多个所述反光工件中各反光工件的原始反光图像作为所述第一生成器的输入，多个所述反光工件中各反光工件的原始无反光图像作为所述第二生成器的输入；其中，所述第一生成器的网络模块和所述第二生成器的网络模块的网络结构相同，包括编码器、辅助分类器和解码器；所述第一判别器的网络模块和所述第二判别器的网络模块的网络结构相同，包括编码器和辅助分类器；所述第一生成器用于基于输入的反光工件的原始反光图像生成所述输入的反光工件的无反光图像，所述第一判别器用于确定所述第一生成器模块生成的所述输入的反光工件的无反光图像、与所述输入的反光工件的原始无反光图像之间的相似度；所述第二生成器用于基于输入的反光工件的原始无反光图像生成所述输入的反光工件的反光图像，所述第二判别器用于确定所述第二生成器模块生成的所述输入的反光工件的反光图像、与所述输入的反光工件的原始反光图像之间的相似度。

5、本技术实施例还提供一种反光擦除装置，包括：图像获取模块，用于获取待反光擦除的目标工件图像；反光检测模块，用于将所述目标工件图像输入至反光检测网络中，以输出所述目标工件图像中的反光检测结果，所述反光检测网络为基于多个反光图像和对应的反光区域标注训练得到的；反光擦除模块，用于基于反光擦除网络和所述目标工件图像的反光检测结果，对所述目标工件图像进行反光擦除操作，得到反光擦除后的目标工件图像，所述反光擦除网络为基于反光擦除网络的训练方法训练得到的。

6、本技术实施例还提供一种电子设备，包括：存储器和处理器；所述存储器，用于存储计算机程序；所述处理器，与所述存储器耦合，用于执行所述计算机程序，以用于实现如反光擦除网络的训练方法或如反光擦除方法中的步骤。

7、本技术实施例还提供一种存储有计算机程序的计算机可读存储介质，当所述计算机程序被处理器执行时，致使所述处理器实现如反光擦除网络的训练方法或如反光擦除方法中的步骤。

8、本技术实施例中提供的反光擦除网络的训练方法，能够获取多个反光工件在多个不同成像效果下的原始反光图像，以及多个反光工件的原始无反光图像；再建立多个反光工件中各反光工件的原始反光图像和原始无反光图像之间的映射关系，以获取反光擦除网络训练集；最后基于建立了各反光工件的原始反光图像和原始无反光图像之间的映射关系的反光擦除网络训练集，训练得到反光擦除网络，该反光擦除网络包括第一生成器、第二生成器、第一判别器和第二判别器，第一生成器与第一判别器连接，第二生成器和第二判别器连接，多个反光工件中各反光工件的原始反光图像作为第一生成器的输入，多个反光工件中各反光工件的原始无反光图像作为第二生成器的输入；其中，所述第一生成器的网络模块和所述第二生成器的网络模块的网络结构相同，包括编码器、辅助分类器和解码器；所述第一判别器的网络模块和所述第二判别器的网络模块的网络结构相同，包括编码器和辅助分类器。由于第一生成器用于基于输入的反光工件的原始反光图像生成输入的反光工件的无反光图像，第一判别器用于确定第一生成器模块生成的输入的反光工件的无反光图像、与输入的反光工件的原始无反光图像之间的相似度，而且第二生成器用于基于输入的反光工件的原始无反光图像生成输入的反光工件的反光图像，第二判别器则用于确定第二生成器模块生成的输入的反光工件的反光图像、与第输入的反光工件的原始反光图像之间的相似度，显然，第一判别器的结果能够用来优化第一生成器的网络参数，第二判别器的结果能够用来优化第二生成器的网络参数，而且第一生成器的输出结果又作为第一判别器的输入，第二生成器的输出结果作为第二判别器的输入，使得第一判别器和第一生成器能够相互改善，第二判别器和第二生成器能够相互改善。以此方法训练得到的反光擦除网络能够准确实现对反光区域的擦除，再对擦除反光区域的工件图像进行缺陷检测便能提高其缺陷检测的准确率。

9、本技术实施例中提供的反光擦除方法，能够获取待反光擦除的目标工件图像，并将目标工件图像输入至反光检测网络中，以输出目标工件图像中的反光检测结果，反光检测网络为基于多个反光图像和对应的反光区域标注训练得到的，最后基于反光擦除网络和目标工件图像的反光检测结果，对目标工件图像进行反光擦除操作，得到反光擦除后的目标工件图像，反光擦除网络为基于反光擦除网络的训练方法训练得到的。由于能够结合反光检测网络和反光擦除网络对存在反光的工件图像进行反光区域的检测以及反光区域的擦除操作，在此基础上再对具备高反光材料的工件表面图像进行缺陷检测，能够极大地提高工件表面的缺陷检测准确率。