本申请实施例公开了一种缺陷图像生成方法、装置、电子设备及存储介质，所述方法包括：获取屏幕正常图像，从屏幕正常图像中确定至少一个目标区域；随机从多个缺陷种类中，确定屏幕正常图像对应的目标缺陷种类；根据目标缺陷种类，生成与各个目标区域对应的缺陷区域；将各个缺陷区域与对应的目标区域进行特征比对；若第一缺陷区域与对应的第一目标区域不同，则将第一目标区域替换成所述第一缺陷区域，以得到屏幕缺陷样本图像。实施该实施例，能够自动生成屏幕缺陷样本图像，并提高了屏幕缺陷样本图像的准确性。图像的准确性。图像的准确性。
【技术实现步骤摘要】
缺陷图像生成方法、装置、电子设备及存储介质
[0001]本申请涉及图像处理
，尤其涉及一种缺陷图像生成方法、装置、电子设备及存储介质。
技术介绍
[0002]屏幕异常缺陷在电子设备中是出现频率较多的一类异常，屏幕异常缺陷对于产品质量的影响较大，严重影响客户对于产品的体验感。在相关技术中，通常采用深度学习算法进行模型训练，通过训练好的模型可以针对不同图像内容中的屏幕异常缺陷进行检测，并且能够达到较好的检测效果。而深度学习算法通常需要提供大量的屏幕缺陷样本图像进行模型训练，拥有更多屏幕缺陷样本图像，越能的提高训练好的模型的检测精度。
[0003]当前，屏幕缺陷样本图像主要在实际生产中进行收集，通过长时间的协调配合，才能收集到足够多的屏幕缺陷样本图像。但随着电子设备的生产工艺水平的不断提高，出现屏幕缺陷样本图像的概率越来越小，屏幕缺陷样本图像收集难度大，收集周期长，导致深度学习算法的训练周期延长，项目整体成本增加。
技术实现思路
[0004]本申请实施例公开了一种缺陷图像生成方法、装置、电子设备及存储介质，能够自动生成屏幕缺陷样本图像，并提高了屏幕缺陷样本图像的准确性。
[0005]本申请实施例公开了一种缺陷图像生成方法，所述方法包括：
[0006]获取屏幕正常图像，从所述屏幕正常图像中确定至少一个目标区域；
[0007]随机从多个缺陷种类中，确定所述屏幕正常图像对应的目标缺陷种类；
[0008]根据所述目标缺陷种类，生成与各个所述目标区域对应的缺陷区域，所述缺陷区域的区域内容与所述目标缺陷种类匹配；
[0009]将各个所述缺陷区域与对应的目标区域进行特征比对；
[0010]若第一缺陷区域与对应的第一目标区域不同，则将所述第一目标区域替换成所述第一缺陷区域，以得到屏幕缺陷样本图像；其中，所述第一缺陷区域为任一所述缺陷区域。
[0011]在一个实施例中，所述多个缺陷种类包括白屏缺陷、黑屏缺陷、彩屏缺陷以及花屏缺陷中的至少两种，所述彩屏缺陷为非黑白的其它单一颜色的缺陷样式，所述花屏缺陷为多种颜色的缺陷样式；
[0012]所述根据所述目标缺陷种类，生成与各个所述目标区域对应的缺陷区域，包括：
[0013]若所述目标缺陷种类为所述白屏缺陷、所述黑屏缺陷以及所述彩屏缺陷中的任一种，则确定与所述目标缺陷种类对应的像素值取值范围；
[0014]随机从像素值取值范围中，确定各个所述目标区域对应的目标像素值；
[0015]根据各个所述目标区域对应的目标像素值，生成与各个所述目标区域对应的缺陷区域；
[0016]若所述目标缺陷种类为所述花屏缺陷，则通过所述花屏缺陷生成模型，生成与各
个目标区域对应的缺陷区域。
[0017]在一个实施例中，所述花屏生成对抗模型是基于生成对抗网络GANomaly模型建立的，所述花屏生成对抗模型包括第一编码器、解码器、判别器以及第二编码器，所述通过所述花屏缺陷生成模型，生成与各个目标区域对应的缺陷区域，包括：
[0018]获取与第二目标区域大小相同的花屏样本图像；所述第二目标区域为所述至少一个目标区域中的任一目标区域；
[0019]通过所述第一编码器对所述花屏样本图像进行卷积特征提取以及特征压缩，得到所述花屏样本图像对应的第一向量；
[0020]通过所述解码器对所述第一向量进行解码，得到所述花屏样本图像对应的重构图像；
[0021]通过所述第二解码器对所述重构图像进行卷积特征提提取以及特征压缩，得到所述重构图像对应的第二向量；
[0022]根据所述第一向量以及第二向量，确定所述花屏样本图像与所述重构图像的相似程度；
[0023]若所述判别器基于所述相似程度确定所述重构图像为真，则将所述重构图像作为所述第一目标区域对应的缺陷图像。
[0024]在一个实施例中，所述从所述屏幕正常图像中确定至少一个目标区域，包括：
[0025]从预设数量范围内随机选取区域数量；
[0026]从所述屏幕正常图像中确定与所述区域数量匹配的目标区域。
[0027]在一个实施例中，所述根据所述区域数量，从所述屏幕正常图像中确定至少一个目标区域，包括：
[0028]根据所述区域数量，对所述屏幕正常图像进行区域等分，得到与所述区域数量匹配的等分区域；
[0029]在每个所述等分区域中随机确定一个目标区域，所述目标区域的区域尺寸小于所述等分区域的区域尺寸。
[0030]在一个实施例中，所述目标区域为矩形，在每个所述等分区域中随机确定一个目标区域，包括：
[0031]随机生成目标等分区域对应的目标区域的第二区域长度及第二区域宽度；其中，所述目标等分区域为任一等分区域，所述第二区域长度小于或等于所述目标等分区域的第一区域长度，所述第二区域宽度小于或等于所述目标等分区域的第一区域宽度；
[0032]根据所述第一区域长度及所述第二区域长度，确定第一坐标取值范围；
[0033]根据所述第一区域宽度及所述第二区域宽度，确定第二坐标取值范围；
[0034]在所述第一坐标取值范围及所述第二坐标取值范围中，随机确定目标顶点的顶点坐标；
[0035]根据所述顶点坐标、所述第二区域长度及所述第二区域宽度，确定所述目标等分区域对应的目标区域。
[0036]在一个实施例中，所述将各个所述缺陷区域与对应的目标区域进行特征比对，包括：
[0037]确定所述第一缺陷区域的第一平均像素值；
[0038]确定所述第一缺陷区域对应的第一目标区域的第二平均像素值；
[0039]计算所述第一平均像素值与所述第二平均像素值之间的目标差值；
[0040]若所述目标差值小于预设阈值，则确定所述第一缺陷区域与所述第一目标区域相同；
[0041]若所述目标差值不小于所述预设阈值，则确定所述第一缺陷区域与所述第一目标区域不同。
[0042]本申请实施例公开了一种缺陷图像生成装置，其特征在于，所述装置包括：
[0043]区域确定模块，用于获取屏幕正常图像，从所述屏幕正常图像中确定至少一个目标区域；
[0044]种类确定模块，用于随机从多个缺陷种类中，确定所述屏幕正常图像对应的目标缺陷种类；
[0045]区域生成模块，用于根据所述目标缺陷种类，生成与各个所述目标区域对应的缺陷区域，所述缺陷区域的区域内容与所述目标缺陷种类匹配；
[0046]特征比对模块，用于将各个所述缺陷区域与对应的目标区域进行特征比对；
[0047]区域替换模块，用于若第一缺陷区域与对应的第一目标区域不同，则将所述第一目标区域替换成所述第一缺陷区域，以得到屏幕缺陷样本图像；其中，所述第一缺陷区域为任一所述缺陷区域。
[0048]在一个实施例中，所述多个缺陷种类包括白屏缺陷、黑屏缺陷、彩屏缺陷以及花屏缺陷中的至少两种，所述彩屏缺陷为非黑白的其它单一颜色的缺陷样式，所述花屏缺陷为多种颜色的缺本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种缺陷图像生成方法，其特征在于，所述方法包括：获取屏幕正常图像，从所述屏幕正常图像中确定至少一个目标区域；随机从多个缺陷种类中，确定所述屏幕正常图像对应的目标缺陷种类；根据所述目标缺陷种类，生成与各个所述目标区域对应的缺陷区域，所述缺陷区域的区域内容与所述目标缺陷种类匹配；将各个所述缺陷区域与对应的目标区域进行特征比对；若第一缺陷区域与对应的第一目标区域不同，则将所述第一目标区域替换成所述第一缺陷区域，以得到屏幕缺陷样本图像；其中，所述第一缺陷区域为任一所述缺陷区域。2.根据权利要求1所述的方法，其特征在于，所述多个缺陷种类包括白屏缺陷、黑屏缺陷、彩屏缺陷以及花屏缺陷中的至少两种，所述彩屏缺陷为非黑白的其它单一颜色的缺陷样式，所述花屏缺陷为多种颜色的缺陷样式；所述根据所述目标缺陷种类，生成与各个所述目标区域对应的缺陷区域，包括：若所述目标缺陷种类为所述白屏缺陷、所述黑屏缺陷以及所述彩屏缺陷中的任一种，则确定与所述目标缺陷种类对应的像素值取值范围；随机从像素值取值范围中，确定各个所述目标区域对应的目标像素值；根据各个所述目标区域对应的目标像素值，生成与各个所述目标区域对应的缺陷区域；若所述目标缺陷种类为所述花屏缺陷，则通过所述花屏缺陷生成模型，生成与各个目标区域对应的缺陷区域。3.根据权利要求2所述的方法，其特征在于，所述花屏生成对抗模型是基于生成对抗网络GANomaly模型建立的，所述花屏生成对抗模型包括第一编码器、解码器、判别器以及第二编码器，所述通过所述花屏缺陷生成模型，生成与各个目标区域对应的缺陷区域，包括：获取与第二目标区域大小相同的花屏样本图像；所述第二目标区域为所述至少一个目标区域中的任一目标区域；通过所述第一编码器对所述花屏样本图像进行卷积特征提取以及特征压缩，得到所述花屏样本图像对应的第一向量；通过所述解码器对所述第一向量进行解码，得到所述花屏样本图像对应的重构图像；通过所述第二解码器对所述重构图像进行卷积特征提提取以及特征压缩，得到所述重构图像对应的第二向量；根据所述第一向量以及第二向量，确定所述花屏样本图像与所述重构图像的相似程度；若所述判别器基于所述相似程度确定所述重构图像为真，则将所述重构图像作为所述第一目标区域对应的缺陷图像。4.根据权利要求1所述的方法，其特征在于，所述从所述屏幕正常图像中确定至少一个目标区域，包括：从预设数量范围内随机选取区域数量；从所述屏幕正常图像中确定与所述区域数量匹配的目标区域。5.根据权利要求4的所述的方法，其特征在于，所述根据所述区域数量，从所述屏幕正常图像中确定至少一个目标...
【专利技术属性】
技术研发人员：阳建峰，
申请(专利权)人：上海闻泰电子科技有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。