图像瑕疵检测方法、装置、电子设备及介质与流程

文档序号:30502385发布日期:2022-06-24 23:13阅读:71来源:国知局
图像瑕疵检测方法、装置、电子设备及介质与流程

1.本技术涉及图像分析技术领域,尤其涉及一种图像瑕疵检测方法、装置、电子设备及介质。


背景技术:

2.在工业检测领域中,对于产品图像的破损、擦伤等缺陷的检测,能够确保产品的高品质。由于瑕疵存在不同尺度的变化,为此,目前的图像瑕疵检测方法无法涵盖各种尺度的瑕疵种类变化,从而降低瑕疵检测的准确度。


技术实现要素:

3.鉴于以上内容,有必要提供一种图像瑕疵检测方法、装置、电子设备及介质,能够提高瑕疵检测的准确度。
4.本技术的第一方面提供一种图像瑕疵检测方法,所述图像瑕疵检测方法包括:
5.当接收到图像瑕疵检测请求时,从所述图像瑕疵检测请求中提取待检测图像;
6.将所述待检测图像转换为多个不同预设尺寸的尺寸图像;
7.利用预先构建的编码器对所述多个尺寸图像进行特征提取,得到多个潜向量;
8.将所述多个潜向量输入至与所述编码器对应的解码器中,得到多个重构图像,及将所述多个潜向量输入至预先训练的高斯混合模型中,得到每个尺寸图像对应的估测概率;
9.将每个重构图像与每个尺寸图像进行比较,得到每个尺寸图像的计算误差;
10.根据每个估测概率及每个计算误差确定每个尺寸图像的总误差,并根据每个总误差及与每个总误差对应的预设阈值确定所述待检测图像的检测结果。
11.根据本技术优选实施例,所述从所述图像瑕疵检测请求中提取待检测图像包括:
12.从线程连接池中获取闲置线程;
13.利用所述闲置线程解析所述图像瑕疵检测请求的方法体,得到所述图像瑕疵检测请求中携带的所有信息;
14.获取预设标签,并根据所述预设标签从所述所有信息中提取与所述预设标签对应的信息作为所述待检测图像。
15.根据本技术优选实施例,所述将所述待检测图像转换为多个不同预设尺寸的尺寸图像包括:
16.根据所述图像瑕疵检测请求确定检测对象;
17.从所述待检测图像中提取包含所述检测对象的区域图像;
18.根据所述不同预设尺寸将所述区域图像转换为所述多个尺寸图像。
19.根据本技术优选实施例,所述利用预先构建的编码器对所述多个尺寸图像进行特征提取,得到多个潜向量包括:
20.对每个尺寸图像进行编码处理,得到每个尺寸图像的特征向量;
21.将每个特征向量输入至所述编码器中的多个隐层进行运算,并将最后一层隐层的输出向量确定为每个尺寸图像对应的潜向量。
22.根据本技术优选实施例,在将所述多个潜向量输入至预先训练的高斯混合模型中,得到每个尺寸图像对应的估测概率之前,所述图像瑕疵检测方法还包括:
23.获取多个训练图像;
24.对所述多个训练图像进行编码,得到多个编码向量,并对所述多个编码向量进行降维并归一化处理,得到多个低维向量;
25.将所述多个低维向量划分为训练集、测试集及验证集;
26.基于最大期望算法,迭代训练所述训练集中的低维向量,得到学习器;
27.利用所述测试集中的低维向量测试所述学习器,得到测试结果;
28.当所述测试结果小于配置值时,利用所述验证集中的低维向量调整所述学习器的参数,得到所述高斯混合模型。
29.根据本技术优选实施例,所述将每个重构图像与每个尺寸图像进行比较,得到每个尺寸图像的计算误差包括:
30.对于任意重构图像,提取所述任意重构图像中的所有重构像素点,并计算所述所有重构像素点的第一数量;
31.提取每个尺寸图像中的所有目标像素点;
32.比较每个重构像素点与每个目标像素点,得到比较结果,其中,将重构像素点与目标像素点不同的比较结果确定为目标比较结果;
33.计算所述目标比较结果的第二数量,并将所述第二数量除以所述第一数量,得到每个尺寸图像的计算误差。
34.根据本技术优选实施例,所述根据每个总误差及与每个总误差对应的预设阈值确定所述待检测图像的检测结果包括:
35.从所述对应的预设阈值中确定最小预设阈值;
36.当每个总误差都小于所述最小预设阈值时,将所述检测结果确定为所述待检测图像无瑕疵;或者
37.当任意总误差大于或者等于所述对应的预设阈值时,将所述检测结果确定为所述待检测图像有瑕疵;或者
38.当每个总误差小于所述对应的预设阈值,且每个总误差大于或者等于所述最小预设阈值时,将所述检测结果确定为所述待检测图像有瑕疵。
39.本技术的第二方面提供一种图像瑕疵检测装置,所述图像瑕疵检测装置包括:
40.提取单元,用于当接收到图像瑕疵检测请求时,从所述图像瑕疵检测请求中提取待检测图像;
41.转换单元,用于将所述待检测图像转换为多个不同预设尺寸的尺寸图像;
42.所述提取单元,还用于利用预先构建的编码器对所述多个尺寸图像进行特征提取,得到多个潜向量;
43.输入单元,用于将所述多个潜向量输入至与所述编码器对应的解码器中,得到多个重构图像,及将所述多个潜向量输入至预先训练的高斯混合模型中,得到每个尺寸图像对应的估测概率;
44.比较单元,用于将每个重构图像与每个尺寸图像进行比较,得到每个尺寸图像的计算误差;
45.确定单元,用于根据每个估测概率及每个计算误差确定每个尺寸图像的总误差,并根据每个总误差及与每个总误差对应的预设阈值确定所述待检测图像的检测结果。
46.根据本技术优选实施例,所述提取单元从所述图像瑕疵检测请求中提取待检测图像包括:
47.从线程连接池中获取闲置线程;
48.利用所述闲置线程解析所述图像瑕疵检测请求的方法体,得到所述图像瑕疵检测请求中携带的所有信息;
49.获取预设标签,并根据所述预设标签从所述所有信息中提取与所述预设标签对应的信息作为所述待检测图像。
50.根据本技术优选实施例,所述转换单元具体用于:
51.根据所述图像瑕疵检测请求确定检测对象;
52.从所述待检测图像中提取包含所述检测对象的区域图像;
53.根据所述不同预设尺寸将所述区域图像转换为所述多个尺寸图像。
54.根据本技术优选实施例,所述提取单元利用预先构建的编码器对所述多个尺寸图像进行特征提取,得到多个潜向量包括:
55.对每个尺寸图像进行编码处理,得到每个尺寸图像的特征向量;
56.将每个特征向量输入至所述编码器中的多个隐层进行运算,并将最后一层隐层的输出向量确定为每个尺寸图像对应的潜向量。
57.根据本技术优选实施例,所述图像瑕疵检测装置还包括:
58.获取单元,用于在将所述多个潜向量输入至预先训练的高斯混合模型中,得到每个尺寸图像对应的估测概率之前,获取多个训练图像;
59.处理单元,用于对所述多个训练图像进行编码,得到多个编码向量,并对所述多个编码向量进行降维并归一化处理,得到多个低维向量;
60.划分单元,用于将所述多个低维向量划分为训练集、测试集及验证集;
61.训练单元,用于基于最大期望算法,迭代训练所述训练集中的低维向量,得到学习器;
62.测试单元,用于利用所述测试集中的低维向量测试所述学习器,得到测试结果;
63.调整单元,用于当所述测试结果小于配置值时,利用所述验证集中的低维向量调整所述学习器的参数,得到所述高斯混合模型。
64.根据本技术优选实施例,所述比较单元具体用于:
65.对于任意重构图像,提取所述任意重构图像中的所有重构像素点,并计算所述所有重构像素点的第一数量;
66.提取每个尺寸图像中的所有目标像素点;
67.比较每个重构像素点与每个目标像素点,得到比较结果,其中,将重构像素点与目标像素点不同的比较结果确定为目标比较结果;
68.计算所述目标比较结果的第二数量,并将所述第二数量除以所述第一数量,得到每个尺寸图像的计算误差。
69.根据本技术优选实施例,所述确定单元根据每个总误差及与每个总误差对应的预设阈值确定所述待检测图像的检测结果包括:
70.从所述对应的预设阈值中确定最小预设阈值;
71.当每个总误差都小于所述最小预设阈值时,将所述检测结果确定为所述待检测图像无瑕疵;或者
72.当任意总误差大于或者等于所述对应的预设阈值时,将所述检测结果确定为所述待检测图像有瑕疵;或者
73.当每个总误差小于所述对应的预设阈值,且每个总误差大于或者等于所述最小预设阈值时,将所述检测结果确定为所述待检测图像有瑕疵。
74.本技术的第三方面提供一种电子设备,所述电子设备包括:
75.存储器,存储至少一个指令;及
76.处理器,获取所述存储器中存储的指令以实现所述图像瑕疵检测方法。
77.本技术的第四方面提供一种计算机可读存储介质,所述计算机可读存储介质中存储有至少一个指令,所述至少一个指令被电子设备中的处理器获取以实现所述图像瑕疵检测方法。
78.由以上技术方案可以看出,本技术通过对待检测图像进行尺寸转换,涵盖了各种尺度的瑕疵种类变化,能够提高瑕疵检测的准确度。
附图说明
79.图1是本技术图像瑕疵检测方法的较佳实施例的流程图。
80.图2是本技术图像瑕疵检测装置的较佳实施例的功能模块图。
81.图3是本技术实现图像瑕疵检测方法的较佳实施例的电子设备的结构示意图。
具体实施方式
82.为了使本技术的目的、技术方案和优点更加清楚,下面结合附图和具体实施例对本技术进行详细描述。
83.如图1所示,是本技术图像瑕疵检测方法的较佳实施例的流程图。根据不同的需求,该流程图中步骤的顺序可以改变,某些步骤可以省略。
84.所述图像瑕疵检测方法应用于一个或者多个电子设备1中,所述电子设备1是一种能够按照事先设定或存储的指令,自动进行数值计算和/或信息处理的设备,其硬件包括但不限于微处理器、专用集成电路(application specific integrated circuit,asic)、可编程门阵列(field-programmable gate array,fpga)、数字信号处理器(digital signal processor,dsp)、嵌入式设备等。
85.所述电子设备1可以是任何一种可与用户进行人机交互的电子产品,例如,个人计算机、平板电脑、智能手机、个人数字助理(personal digital assistant,pda)、游戏机、交互式网络电视(internet protocol television,iptv)、智能式穿戴式设备等。
86.所述电子设备1还可以包括网络设备和/或用户设备。其中,所述网络设备包括,但不限于单个网络服务器、多个网络服务器组成的服务器组或基于云计算(cloud computing)的由大量主机或网络服务器构成的云。
87.所述电子设备1所处的网络包括但不限于互联网、广域网、城域网、局域网、虚拟专用网络(virtual private network,vpn)等。
88.s10,当接收到图像瑕疵检测请求时,从所述图像瑕疵检测请求中提取待检测图像。
89.在本技术的至少一个实施例中,所述图像瑕疵检测请求可以由用户触发(例如:通过预设功能按键进行触发),也可以在预设时间内自动触发,本技术不作限制。
90.其中,所述预设时间可以是时间点(例如:每天早上九点),也可以是时间段。
91.在本技术的至少一个实施例中,所述图像瑕疵检测请求中携带的信息包括,但不限于:检测对象、待检测图像。可以理解的是,所述待检测图像中包含所述检测对象。
92.在本技术的至少一个实施例中,所述电子设备从所述图像瑕疵检测请求中提取待检测图像包括:
93.所述电子设备从线程连接池中获取闲置线程,进一步地,所述电子设备利用所述闲置线程解析所述图像瑕疵检测请求的方法体,得到所述图像瑕疵检测请求中携带的所有信息,更进一步地,所述电子设备获取预设标签,并根据所述预设标签从所述所有信息中提取与所述预设标签对应的信息作为所述待检测图像。
94.其中,所述线程连接池中存储多个线程;进一步地,所述预设标签是指用于标识所述待检测图像的标签。
95.通过从所述线程连接池中获取闲置线程,能够节省线程的创建时间,进而通过解析所述图像瑕疵检测请求的方法体,能够缩短所述图像瑕疵检测请求的解析时长,进而通过预设标签与待检测图像的映射关系,能够准确确定所述待检测图像。
96.s11,将所述待检测图像转换为多个不同预设尺寸的尺寸图像。
97.在本技术的至少一个实施例中,所述多个不同预设尺寸可以包括,但不限于:32*32、64*64、128*128。
98.在本技术的至少一个实施例中,所述电子设备将所述待检测图像转换为多个不同预设尺寸的尺寸图像包括:
99.所述电子设备根据所述图像瑕疵检测请求确定所述检测对象,所述电子设备从所述待检测图像中提取包含所述检测对象的区域图像,进一步地,所述电子设备根据所述不同预设尺寸将所述区域图像转换为所述多个尺寸图像。
100.通过上述实施方式,能够确保转换后的多个尺寸图像中包含检测对象,另外,由于只需对包含检测对象的区域图像进行尺寸转换,而无需对待检测图像进行尺寸转换,减少了尺寸转换前的图像像素点,能够提高尺寸转换效率。
101.具体地,所述电子设备可以利用开源的opencv工具将所述区域图像转换为具有所述不同预设尺寸的所述多个尺寸图像。
102.s12,利用预先构建的编码器对所述多个尺寸图像进行特征提取,得到多个潜向量。
103.在本技术的至少一个实施例中,所述编码器可以是自编码器(autoencoder,ae)中的编码器。进一步地,所述编码器中包含多个隐层,所述多个隐层的数量可以根据应用场景任意设置。
104.在本技术的至少一个实施例中,所述电子设备利用预先构建的编码器对所述多个
尺寸图像进行特征提取,得到多个潜向量包括:
105.所述电子设备对每个尺寸图像进行编码处理,得到每个尺寸图像的特征向量,进一步地,所述电子设备将每个特征向量输入至所述编码器中的多个隐层进行运算,并将最后一层隐层的输出向量确定为每个尺寸图像对应的潜向量。
106.具体地,所述电子设备将每个特征向量输入至所述编码器中的多个隐层进行运算包括:
107.对于任意隐层,所述电子设备获取所述任意隐层对应的权重矩阵及偏置值,进一步地,所述电子设备将每个特征向量与所述权重矩阵进行相乘运算,得到运算结果,并将所述运算结果加上所述偏置值,得到所述任意隐层的输出向量。
108.s13,将所述多个潜向量输入至与所述编码器对应的解码器中,得到多个重构图像,及将所述多个潜向量输入至预先训练的高斯混合模型中,得到每个尺寸图像对应的估测概率。
109.在本技术的至少一个实施例中,所述解码器可以是自编码器(autoencoder,ae)中的解码器。进一步地,所述解码器中包含多个与所述编码器中的隐层对应的运算层,所述多个运算层的数量与所述多个隐层的数量相同。
110.在本技术的至少一个实施例中,所述高斯混合模型(gaussian mixture model,gmm)能够利用高斯概率密度函数(正态分布曲线)精确地量化所述多个潜向量。
111.在本技术的至少一个实施例中,在将所述多个潜向量输入至预先训练的高斯混合模型中,得到每个尺寸图像对应的估测概率之前,所述图像瑕疵检测方法还包括:
112.所述电子设备获取多个训练图像,进一步地,所述电子设备对所述多个训练图像进行编码,得到多个编码向量,并对所述多个编码向量进行降维并归一化处理,得到多个低维向量,更进一步地,所述电子设备将所述多个低维向量划分为训练集、测试集及验证集,更进一步地,所述电子设备基于最大期望算法,迭代训练所述训练集中的低维向量,得到学习器,更进一步地,所述电子设备利用所述测试集中的低维向量测试所述学习器,得到测试结果,当所述测试结果小于配置值时,所述电子设备利用所述验证集中的低维向量调整所述学习器的参数,得到所述高斯混合模型。
113.通过上述实施方式,能够使生成的高斯混合模型更加准确。
114.在本技术的至少一个实施例中,在将所述多个低维向量划分为训练集、测试集及验证集之前,所述方法还包括:
115.所述电子设备计算所述多个低维向量的数量,当所述数量小于预设数量时,所述电子设备利用数据增强算法增加所述多个低维向量的数量。
116.通过上述实施方式,能够避免由于多个低维向量的数量不足,导致训练得到的高斯混合模型生成的估测概率的泛化能力较差。
117.在本技术的至少一个实施例中,所述电子设备将所述多个低维向量划分为训练集、测试集及验证集包括:
118.所述电子设备将所述多个低维向量按照预设比例随机划分为多个数据包,将所述多个数据包中的任意两个数据包分别确定为所述验证集及所述测试集,其余的数据包确定为所述训练集,重复上述步骤,直至所有的数据包全都依次被用作为所述验证集。
119.其中,所述预设比例可以自定义设置,本技术不作限制。
120.通过上述实施方式,使所述多个低维向量中的每个低维向量均参与训练及验证,由此,提高训练所述高斯混合模型的拟合度。
121.s14,将每个重构图像与每个尺寸图像进行比较,得到每个尺寸图像的计算误差。
122.在本技术的至少一个实施例中,所述电子设备将每个重构图像与每个尺寸图像进行比较,得到每个尺寸图像的计算误差包括:
123.对于任意重构图像,所述电子设备提取所述任意重构图像中的所有重构像素点,并计算所述所有重构像素点的第一数量,进一步地,所述电子设备提取每个尺寸图像中的所有目标像素点,所述电子设备比较每个重构像素点与每个目标像素点,得到比较结果,其中,将重构像素点与目标像素点不同的比较结果确定为目标比较结果,更进一步地,所述电子设备计算所述目标比较结果的第二数量,并将所述第二数量除以所述第一数量,得到每个尺寸图像的计算误差。
124.通过上述实施方式,能够准确确定每个尺寸图像的计算误差。
125.s15,根据每个估测概率及每个计算误差确定每个尺寸图像的总误差,并根据每个总误差及与每个总误差对应的预设阈值确定所述待检测图像的检测结果。
126.在本技术的至少一个实施例中,所述检测结果包括所述待检测图像有瑕疵及所述待检测图像无瑕疵。
127.在本技术的至少一个实施例中,所述电子设备根据每个估测概率及每个计算误差确定每个尺寸图像的总误差包括:
128.所述电子设备获取与估测概率对应的第一预设权重,并获取与计算误差对应的第二预设权重,所述第一预设权重与所述第二预设权重之和为1,进一步地,所述电子设备将每个估测概率乘以所述第一预设权重,得到第一数值,并将每个计算误差乘以所述第二预设权重,得到第二数值,更进一步地,所述电子设备将每个第一数值与每个第二数值相加,得到每个尺寸图像的总误差。
129.通过上述实施方式,能够快速确定每个尺寸图像的总误差。
130.在本技术的至少一个实施例中,所述电子设备根据每个总误差及与每个总误差对应的预设阈值确定所述待检测图像的检测结果包括:
131.所述电子设备从所述对应的预设阈值中确定最小预设阈值,当每个总误差都小于所述最小预设阈值时,所述电子设备将所述检测结果确定为所述待检测图像无瑕疵,或者,当任意总误差大于或者等于所述对应的预设阈值时,所述电子设备将所述检测结果确定为所述待检测图像有瑕疵,或者,当每个总误差小于所述对应的预设阈值,且每个总误差大于或者等于所述最小预设阈值时,所述电子设备将所述检测结果确定为所述待检测图像有瑕疵。
132.在本技术的至少一个实施例中,当所述待检测图像有瑕疵时,所述电子设备根据所述待检测图像生成提醒信息,并将所述提醒信息发送至指定联系人的终端设备中。
133.由以上技术方案可以看出,本技术通过对待检测图像进行尺寸转换,涵盖了各种尺度的瑕疵种类变化,能够提高瑕疵检测的准确度。
134.如图2所示,是本技术图像瑕疵检测装置的较佳实施例的功能模块图。所述图像瑕疵检测装置11包括提取单元110、转换单元111、输入单元112、比较单元113、确定单元114、获取单元115、处理单元116、划分单元117、训练单元118、测试单元119、调整单元120、计算
单元121及生成单元122。本技术所称的模块/单元是指一种能够被处理器13所获取,并且能够完成固定功能的一系列计算机程序段,其存储在存储器12中。在本实施例中,关于各模块/单元的功能将在后续的实施例中详述。
135.当接收到图像瑕疵检测请求时,提取单元110从所述图像瑕疵检测请求中提取待检测图像。
136.在本技术的至少一个实施例中,所述图像瑕疵检测请求可以由用户触发(例如:通过预设功能按键进行触发),也可以在预设时间内自动触发,本技术不作限制。
137.其中,所述预设时间可以是时间点(例如:每天早上九点),也可以是时间段。
138.在本技术的至少一个实施例中,所述图像瑕疵检测请求中携带的信息包括,但不限于:检测对象、待检测图像。可以理解的是,所述待检测图像中包含所述检测对象。
139.在本技术的至少一个实施例中,所述提取单元110从所述图像瑕疵检测请求中提取待检测图像包括:
140.所述提取单元110从线程连接池中获取闲置线程,进一步地,所述提取单元110利用所述闲置线程解析所述图像瑕疵检测请求的方法体,得到所述图像瑕疵检测请求中携带的所有信息,更进一步地,所述提取单元110获取预设标签,并根据所述预设标签从所述所有信息中提取与所述预设标签对应的信息作为所述待检测图像。
141.其中,所述线程连接池中存储多个线程;进一步地,所述预设标签是指用于标识所述待检测图像的标签。
142.通过从所述线程连接池中获取闲置线程,能够节省线程的创建时间,进而通过解析所述图像瑕疵检测请求的方法体,能够缩短所述图像瑕疵检测请求的解析时长,进而通过预设标签与待检测图像的映射关系,能够准确确定所述待检测图像。
143.转换单元111将所述待检测图像转换为多个不同预设尺寸的尺寸图像。
144.在本技术的至少一个实施例中,所述多个不同预设尺寸可以包括,但不限于:32*32、64*64、128*128。
145.在本技术的至少一个实施例中,所述转换单元111将所述待检测图像转换为多个不同预设尺寸的尺寸图像包括:
146.所述转换单元111根据所述图像瑕疵检测请求确定所述检测对象,所述转换单元111从所述待检测图像中提取包含所述检测对象的区域图像,进一步地,所述转换单元111根据所述不同预设尺寸将所述区域图像转换为所述多个尺寸图像。
147.通过上述实施方式,能够确保转换后的多个尺寸图像中包含检测对象,另外,由于只需对包含检测对象的区域图像进行尺寸转换,而无需对待检测图像进行尺寸转换,减少了尺寸转换前的图像像素点,能够提高尺寸转换效率。
148.具体地,所述转换单元111可以利用开源的opencv工具将所述区域图像转换为具有所述不同预设尺寸的所述多个尺寸图像。
149.所述提取单元110利用预先构建的编码器对所述多个尺寸图像进行特征提取,得到多个潜向量。
150.在本技术的至少一个实施例中,所述编码器可以是自编码器(autoencoder,ae)中的编码器。进一步地,所述编码器中包含多个隐层,所述多个隐层的数量可以根据应用场景任意设置。
151.在本技术的至少一个实施例中,所述提取单元110利用预先构建的编码器对所述多个尺寸图像进行特征提取,得到多个潜向量包括:
152.所述提取单元110对每个尺寸图像进行编码处理,得到每个尺寸图像的特征向量,进一步地,所述提取单元110将每个特征向量输入至所述编码器中的多个隐层进行运算,并将最后一层隐层的输出向量确定为每个尺寸图像对应的潜向量。
153.具体地,所述提取单元110将每个特征向量输入至所述编码器中的多个隐层进行运算包括:
154.对于任意隐层,所述提取单元110获取所述任意隐层对应的权重矩阵及偏置值,进一步地,所述提取单元110将每个特征向量与所述权重矩阵进行相乘运算,得到运算结果,并将所述运算结果加上所述偏置值,得到所述任意隐层的输出向量。
155.输入单元112将所述多个潜向量输入至与所述编码器对应的解码器中,得到多个重构图像,及将所述多个潜向量输入至预先训练的高斯混合模型中,得到每个尺寸图像对应的估测概率。
156.在本技术的至少一个实施例中,所述解码器可以是自编码器(autoencoder,ae)中的解码器。进一步地,所述解码器中包含多个与所述编码器中的隐层对应的运算层,所述多个运算层的数量与所述多个隐层的数量相同。
157.在本技术的至少一个实施例中,所述高斯混合模型(gaussian mixture model,gmm)能够利用高斯概率密度函数(正态分布曲线)精确地量化所述多个潜向量。
158.在本技术的至少一个实施例中,在将所述多个潜向量输入至预先训练的高斯混合模型中,得到每个尺寸图像对应的估测概率之前,获取单元115获取多个训练图像,进一步地,处理单元116对所述多个训练图像进行编码,得到多个编码向量,并对所述多个编码向量进行降维并归一化处理,得到多个低维向量,更进一步地,划分单元117将所述多个低维向量划分为训练集、测试集及验证集,更进一步地,训练单元118基于最大期望算法,迭代训练所述训练集中的低维向量,得到学习器,更进一步地,测试单元119利用所述测试集中的低维向量测试所述学习器,得到测试结果,当所述测试结果小于配置值时,调整单元120利用所述验证集中的低维向量调整所述学习器的参数,得到所述高斯混合模型。
159.通过上述实施方式,能够使生成的高斯混合模型更加准确。
160.在本技术的至少一个实施例中,在将所述多个低维向量划分为训练集、测试集及验证集之前,计算单元121计算所述多个低维向量的数量,当所述数量小于预设数量时,所述计算单元121利用数据增强算法增加所述多个低维向量的数量。
161.通过上述实施方式,能够避免由于多个低维向量的数量不足,导致训练得到的高斯混合模型生成的估测概率的泛化能力较差。
162.在本技术的至少一个实施例中,所述划分单元117将所述多个低维向量划分为训练集、测试集及验证集包括:
163.所述划分单元117将所述多个低维向量按照预设比例随机划分为多个数据包,进一步地,所述划分单元117将所述多个数据包中的任意两个数据包分别确定为所述验证集及所述测试集,其余的数据包确定为所述训练集,重复上述步骤,直至所有的数据包全都依次被用作为所述验证集。
164.其中,所述预设比例可以自定义设置,本技术不作限制。
165.通过上述实施方式,使所述多个低维向量中的每个低维向量均参与训练及验证,由此,提高训练所述高斯混合模型的拟合度。
166.比较单元113将每个重构图像与每个尺寸图像进行比较,得到每个尺寸图像的计算误差。
167.在本技术的至少一个实施例中,所述比较单元113将每个重构图像与每个尺寸图像进行比较,得到每个尺寸图像的计算误差包括:
168.对于任意重构图像,所述比较单元113提取所述任意重构图像中的所有重构像素点,并计算所述所有重构像素点的第一数量,进一步地,所述比较单元113提取每个尺寸图像中的所有目标像素点,所述比较单元113比较每个重构像素点与每个目标像素点,得到比较结果,其中,将重构像素点与目标像素点不同的比较结果确定为目标比较结果,更进一步地,所述目标比较单元113计算的第二数量,并将所述第二数量除以所述第一数量,得到每个尺寸图像的计算误差。
169.通过上述实施方式,能够准确确定每个尺寸图像的计算误差。
170.确定单元114根据每个估测概率及每个计算误差确定每个尺寸图像的总误差,并根据每个总误差及与每个总误差对应的预设阈值确定所述待检测图像的检测结果。
171.在本技术的至少一个实施例中,所述检测结果包括所述待检测图像有瑕疵及所述待检测图像无瑕疵。
172.在本技术的至少一个实施例中,所述确定单元114根据每个估测概率及每个计算误差确定每个尺寸图像的总误差包括:
173.所述确定单元114获取与估测概率对应的第一预设权重,并获取与计算误差对应的第二预设权重,所述第一预设权重与所述第二预设权重之和为1,进一步地,所述确定单元114将每个估测概率乘以所述第一预设权重,得到第一数值,并将每个计算误差乘以所述第二预设权重,得到第二数值,更进一步地,所述确定单元114将每个第一数值与每个第二数值相加,得到每个尺寸图像的总误差。
174.通过上述实施方式,能够快速确定每个尺寸图像的总误差。
175.在本技术的至少一个实施例中,所述确定单元114根据每个总误差及与每个总误差对应的预设阈值确定所述待检测图像的检测结果包括:
176.所述确定单元114从所述对应的预设阈值中确定最小预设阈值,当每个总误差都小于所述最小预设阈值时,所述确定单元114将所述检测结果确定为所述待检测图像无瑕疵,或者,当任意总误差大于或者等于所述对应的预设阈值时,所述确定单元114将所述检测结果确定为所述待检测图像有瑕疵,或者,当每个总误差小于所述对应的预设阈值,且每个总误差大于或者等于所述最小预设阈值时,所述确定单元114将所述检测结果确定为所述待检测图像有瑕疵。
177.在本技术的至少一个实施例中,当所述待检测图像有瑕疵时,生成单元122根据所述待检测图像生成提醒信息,并将所述提醒信息发送至指定联系人的终端设备中。
178.由以上技术方案可以看出,本技术通过对待检测图像进行尺寸转换,涵盖了各种尺度的瑕疵种类变化,能够提高瑕疵检测的准确度。
179.如图3所示,是本技术实现图像瑕疵检测方法的较佳实施例的电子设备的结构示意图。
180.在本技术的一个实施例中,所述电子设备1包括,但不限于,存储器12、处理器13,以及存储在所述存储器12中并可在所述处理器13上运行的计算机程序,例如图像瑕疵检测程序。
181.本领域技术人员可以理解,所述示意图仅仅是电子设备1的示例,并不构成对电子设备1的限定,可以包括比图示更多或更少的部件,或者组合某些部件,或者不同的部件,例如所述电子设备1还可以包括输入输出设备、网络接入设备、总线等。
182.所述处理器13可以是中央处理单元(central processing unit,cpu),还可以是其他通用处理器、数字信号处理器(digital signal processor,dsp)、专用集成电路(application specific integrated circuit,asic)、现场可编程门阵列(field-programmable gate array,fpga)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等,所述处理器13是所述电子设备1的运算核心和控制中心,利用各种接口和线路连接整个电子设备1的各个部分,及获取所述电子设备1的操作系统以及安装的各类应用程序、程序代码等。
183.所述处理器13获取所述电子设备1的操作系统以及安装的各类应用程序。所述处理器13获取所述应用程序以实现上述各个图像瑕疵检测方法实施例中的步骤,例如图1所示的步骤。
184.示例性的,所述计算机程序可以被分割成一个或多个模块/单元,所述一个或者多个模块/单元被存储在所述存储器12中,并由所述处理器13获取,以完成本技术。所述一个或多个模块/单元可以是能够完成特定功能的一系列计算机程序指令段,该指令段用于描述所述计算机程序在所述电子设备1中的获取过程。例如,所述计算机程序可以被分割成提取单元110、转换单元111、输入单元112、比较单元113、确定单元114、获取单元115、处理单元116、划分单元117、训练单元118、测试单元119、调整单元120、计算单元121及生成单元122。
185.所述存储器12可用于存储所述计算机程序和/或模块,所述处理器13通过运行或获取存储在所述存储器12内的计算机程序和/或模块,以及调用存储在存储器12内的数据,实现所述电子设备1的各种功能。所述存储器12可主要包括存储程序区和存储数据区,其中,存储程序区可存储操作系统、至少一个功能所需的应用程序(比如声音播放功能、图像播放功能等)等;存储数据区可存储根据电子设备的使用所创建的数据等。此外,存储器12可以包括非易失性存储器,例如硬盘、内存、插接式硬盘,智能存储卡(smart media card,smc),安全数字(secure digital,sd)卡,闪存卡(flash card)、至少一个磁盘存储器件、闪存器件、或其他非易失性固态存储器件。
186.所述存储器12可以是电子设备1的外部存储器和/或内部存储器。进一步地,所述存储器12可以是具有实物形式的存储器,如内存条、tf卡(trans-flash card)等等。
187.所述电子设备1集成的模块/单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本技术实现上述实施例方法中的全部或部分流程,也可以通过计算机程序来指令相关的硬件来完成,所述的计算机程序可存储于一计算机可读存储介质中,该计算机程序在被处理器获取时,可实现上述各个方法实施例的步骤。
188.其中,所述计算机程序包括计算机程序代码,所述计算机程序代码可以为源代码形式、对象代码形式、可获取文件或某些中间形式等。所述计算机可读介质可以包括:能够携带所述计算机程序代码的任何实体或装置、记录介质、u盘、移动硬盘、磁碟、光盘、计算机存储器、只读存储器(rom,read-only memory)。
189.结合图1,所述电子设备1中的所述存储器12存储多个指令以实现一种图像瑕疵检测方法,所述处理器13可获取所述多个指令从而实现:当接收到图像瑕疵检测请求时,从所述图像瑕疵检测请求中提取待检测图像;将所述待检测图像转换为多个不同预设尺寸的尺寸图像;利用预先构建的编码器对所述多个尺寸图像进行特征提取,得到多个潜向量;将所述多个潜向量输入至与所述编码器对应的解码器中,得到多个重构图像,及将所述多个潜向量输入至预先训练的高斯混合模型中,得到每个尺寸图像对应的估测概率;将每个重构图像与每个尺寸图像进行比较,得到每个尺寸图像的计算误差;根据每个估测概率及与每个总误差每个计算误差确定每个尺寸图像的总误差,并根据每个总误差及对应的预设阈值确定所述待检测图像的检测结果。
190.具体地,所述处理器13对上述指令的具体实现方法可参考图1对应实施例中相关步骤的描述,在此不赘述。
191.在本技术所提供的几个实施例中,应该理解到,所揭露的系统,装置和方法,可以通过其它的方式实现。例如,以上所描述的装置实施例仅仅是示意性的,例如,所述模块的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式。
192.所述作为分离部件说明的模块可以是或者也可以不是物理上分开的,作为模块显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。
193.另外,在本技术各个实施例中的各功能模块可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。上述集成的单元既可以采用硬件的形式实现,也可以采用硬件加软件功能模块的形式实现。
194.因此,无论从哪一点来看,均应将实施例看作是示范性的,而且是非限制性的,本技术的范围由所附权利要求而不是上述说明限定,因此旨在将落在权利要求的等同要件的含义和范围内的所有变化涵括在本技术内。不应将权利要求中的任何附关联图标记视为限制所涉及的权利要求。
195.此外,显然“包括”一词不排除其他单元或步骤,单数不排除复数。本技术中陈述的多个单元或装置也可以由一个单元或装置通过软件或者硬件来实现。第一、第二等词语用来表示名称,而并不表示任何特定的顺序。
196.最后应说明的是,以上实施例仅用以说明本技术的技术方案而非限制,尽管参照较佳实施例对本技术进行了详细说明,本领域的普通技术人员应当理解,可以对本技术的技术方案进行修改或等同替换,而不脱离本技术技术方案的精神和范围。
当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1