CN114565567B - 复杂纹理花边布的缺陷检测方法及装置 - Google Patents

复杂纹理花边布的缺陷检测方法及装置 Download PDF

Info

Publication number
CN114565567B
CN114565567B CN202210137881.6A CN202210137881A CN114565567B CN 114565567 B CN114565567 B CN 114565567B CN 202210137881 A CN202210137881 A CN 202210137881A CN 114565567 B CN114565567 B CN 114565567B
Authority
CN
China
Prior art keywords
image
sub
detected
restored
sample
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Active
Application number
CN202210137881.6A
Other languages
English (en)
Other versions
CN114565567A (zh
Inventor
黄必清
许丁
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Tsinghua University
Original Assignee
Tsinghua University
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Tsinghua University filed Critical Tsinghua University
Priority to CN202210137881.6A priority Critical patent/CN114565567B/zh
Publication of CN114565567A publication Critical patent/CN114565567A/zh
Application granted granted Critical
Publication of CN114565567B publication Critical patent/CN114565567B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/0002Inspection of images, e.g. flaw detection
    • G06T7/0004Industrial image inspection
    • G06T7/0006Industrial image inspection using a design-rule based approach
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T9/00Image coding
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/30Subject of image; Context of image processing
    • G06T2207/30108Industrial image inspection
    • G06T2207/30124Fabrics; Textile; Paper
    • YGENERAL TAGGING OF NEW TECHNOLOGICAL DEVELOPMENTS; GENERAL TAGGING OF CROSS-SECTIONAL TECHNOLOGIES SPANNING OVER SEVERAL SECTIONS OF THE IPC; TECHNICAL SUBJECTS COVERED BY FORMER USPC CROSS-REFERENCE ART COLLECTIONS [XRACs] AND DIGESTS
    • Y02TECHNOLOGIES OR APPLICATIONS FOR MITIGATION OR ADAPTATION AGAINST CLIMATE CHANGE
    • Y02PCLIMATE CHANGE MITIGATION TECHNOLOGIES IN THE PRODUCTION OR PROCESSING OF GOODS
    • Y02P90/00Enabling technologies with a potential contribution to greenhouse gas [GHG] emissions mitigation
    • Y02P90/30Computing systems specially adapted for manufacturing

Landscapes

  • Engineering & Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Theoretical Computer Science (AREA)
  • Quality & Reliability (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Multimedia (AREA)
  • Image Analysis (AREA)
  • Investigating Materials By The Use Of Optical Means Adapted For Particular Applications (AREA)
  • Image Processing (AREA)

Abstract

本发明提供一种复杂纹理花边布的缺陷检测方法及装置,获取待检测子图像;待检测子图像中包含花边布图案;确定待检测子图像在目标纹理图像中的第一对应位置,将第一对应位置处的图像和待检测子图像组成待检测图像对;目标纹理图像为花边布图案无缺陷的图像;将待检测图像对输入至编解码模型中,得到编解码模型输出的恢复后的待检测子图像;将恢复后的待检测子图像和目标纹理图像进行比对;在确定恢复后的待检测子图像的花边布图案有缺陷时,确定恢复后的待检测子图像中的缺陷位置。本发明提供的复杂纹理花边布的缺陷检测方法及装置,实现了图像缺陷位置的自动检测,从而提高了图像缺陷检测的效率。

Description

复杂纹理花边布的缺陷检测方法及装置
技术领域
本发明涉及图像技术领域,尤其涉及一种复杂纹理花边布的缺陷检测方法及装置。
背景技术
花边布的表面缺陷会极大影响产品的质量,降低产品的价格。缺陷检测是花边布生产过程中的重要过程,而随着生产过程的推进,花边布的稳定性、规范性会下降,流水线也会开始分化,而且越接近生产后期,缺陷检测难度就越大,且会增加后续的加工成本;因此,有必要在生产流水线中增加一种可靠的缺陷检测手段。
相关技术中,通常采用人工检测的方法,由一部分员工专门负责纺织品缺陷检测工作。
但上述相关技术中,由于花边布的缺陷检测要求高,若通过人工对大量花边布进行缺陷检测,则会降低缺陷检测的效率。
发明内容
针对现有技术存在的问题,本发明提供一种复杂纹理花边布的缺陷检测方法及装置。
本发明提供一种复杂纹理花边布的缺陷检测方法,包括:
获取待检测子图像;其中,所述待检测子图像中包含花边布图案;
确定所述待检测子图像在目标纹理图像中的第一对应位置,将所述第一对应位置处的图像和所述待检测子图像组成待检测图像对;其中,所述目标纹理图像为花边布图案无缺陷的图像;
将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像;
将所述恢复后的待检测子图像和所述目标纹理图像进行比对;
在确定所述恢复后的待检测子图像的花边布图案有缺陷时,确定所述恢复后的待检测子图像中的缺陷位置;
其中,所述编解码模型是基于多个图像对样本训练得到的;所述编解码模型用于基于所述第一对应位置处的图像对所述待检测子图像进行矫正。
根据本发明提供的一种复杂纹理花边布的缺陷检测方法,所述将所述恢复后的待检测子图像和所述目标纹理图像进行比对,包括:
将所述恢复后的待检测子图像和所述目标纹理图像,从亮度、对比度和结构进行结构相似性评估。
根据本发明提供的一种复杂纹理花边布的缺陷检测方法,在所述将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像之前,所述方法还包括:
获取多个时间周期的图像样本序列;其中,所述图像样本序列中至少包括一个时间周期的所述目标纹理图像;所述时间周期为生产流水线中花边布图案循环一次所需的时间;
在所述图像样本序列中提取预设尺寸的多个子图像样本;
确定每个所述子图像样本在所述目标纹理图像中的第二对应位置,将每个所述第二对应位置处的图像和对应的所述子图像样本组成多个图像对样本;
基于所述多个图像对样本对深度学习模型进行训练,得到所述编解码模型。
根据本发明提供的一种复杂纹理花边布的缺陷检测方法,所述基于所述多个图像对样本对深度学习模型进行训练,得到所述编解码模型,包括:
将所述多个图像对样本输入至深度学习模型中,得到每个所述图像对样本对应的光流向量场;
将所述光流向量场与对应的所述子图像样本进行叠加,并对叠加结果进行插值,得到恢复后的子图像样本;
基于所述恢复后的子图像样本和所述目标纹理图像构建损失函数;
基于所述损失函数对所述深度学习模型进行优化,直至所述损失函数收敛,得到所述编解码模型。
根据本发明提供的一种复杂纹理花边布的缺陷检测方法,所述基于所述恢复后的子图像样本和所述目标纹理图像构建损失函数,包括:
采用以下公式(1)构建所述损失函数;
L=Lmse+λLflow (1)
其中,L表示损失函数,λ表示光流向量场梯度损失的权重,Lmse表示所述恢复结果形态损失,Lmse=∑i,j[m′(i,j)-f(i,j)]2;Lflow表示所述光流向量场梯度损失,Lflow=∑i,j[g(i,j,0)2+g(i,j,1)]2;m′(i,j)为所述恢复后的子图像样本,f(i,j)为所述目标纹理图像;g(i,j,0)为第一层的光流向量场,g(i,j,1)为第二层的光流向量场。
根据本发明提供的一种复杂纹理花边布的缺陷检测方法,所述确定每个所述子图像样本在所述目标纹理图像中的第二对应位置,包括:
针对每个所述子图像样本,对所述子图像样本构建尺度空间;
在所述尺度空间中,确定预设邻域中的极值点;
将主曲率比值大于预设阈值的极值点进行过滤;
确定过滤后剩余的每个极值点的特征信息;
基于过滤后剩余的每个极值点的特征信息确定所述子图像样本在所述目标纹理图像中的所述第二对应位置。
根据本发明提供的一种复杂纹理花边布的缺陷检测方法,所述深度学习模型包括4层U-Net网络,所述U-Net网络包括一个输入I、一个输出OUT、中间量C1、中间量C2、中间量C3、中间量C4、中间量C5和中间量C6
所述C1=Conv(ω0I,stride=2);
所述C2=Conv(ω1C1,stride=2);
所述C3=Conv(ω2C2,stride=2);
所述C4=Conv(ω3C3,stride=1);
所述C5=Up(Concat(C2,C4));
所述C6=Up(Concat(C1,C5));
所述OUT=Up(Concat(C1,C5));
其中,Conv(I,stride)表示输入为I和步长stride的二维卷积操作,ω表示卷积层的参数,Concat表示两个图像的拼接操作,Up表示输入图像的上采样操作。
本发明还提供一种复杂纹理花边布的缺陷检测装置,包括:
第一获取单元,用于获取待检测子图像;其中,所述待检测子图像中包含花边布图案;
第一确定单元,用于确定所述待检测子图像在目标纹理图像中的第一对应位置,将所述第一对应位置处的图像和所述待检测子图像组成待检测图像对;其中,所述目标纹理图像为花边布图案无缺陷的图像;
处理单元,用于将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像;
比对单元,用于将所述恢复后的待检测子图像和所述目标纹理图像进行比对;
第二确定单元,用于在确定所述恢复后的待检测子图像的花边布图案有缺陷时,确定所述恢复后的待检测子图像中的缺陷位置;
其中,所述编解码模型是基于多个图像对样本训练得到的;所述编解码模型用于基于所述第一对应位置处的图像对所述待检测子图像进行矫正。
本发明还提供一种电子设备,包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,所述处理器执行所述程序时实现如上述任一种所述复杂纹理花边布的缺陷检测方法的步骤。
本发明还提供一种非暂态计算机可读存储介质,其上存储有计算机程序,该计算机程序被处理器执行时实现如上述任一种所述复杂纹理花边布的缺陷检测方法的步骤。
本发明还提供一种计算机程序产品,包括计算机程序,所述计算机程序被处理器执行时实现如上述任一种所述复杂纹理花边布的缺陷检测方法的步骤。
本发明提供的一种复杂纹理花边布的缺陷检测方法及装置,将确定的待检测图像对输入至编解码模型中,得到编解码模型输出的恢复后的待检测子图像,然后将恢复后的待检测子图像和花边布图案无缺陷的目标纹理图像进行比对,确定出恢复后的待检测子图像中的缺陷位置,实现了图像缺陷位置的自动检测,从而提高了图像缺陷检测的效率。
附图说明
为了更清楚地说明本发明或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作一简单地介绍,显而易见地,下面描述中的附图是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
图1是本发明提供的复杂纹理花边布的缺陷检测方法的流程示意图之一;
图2是本发明提供的复杂纹理花边布的缺陷检测方法的流程示意图之二;
图3是本发明提供的复杂纹理花边布的缺陷检测方法的流程示意图之三;
图4是本发明提供的复杂纹理花边布的缺陷检测方法的流程示意图之四;
图5是本发明提供的复杂纹理花边布的缺陷检测装置的结构示意图;
图6是本发明提供的电子设备的实体结构示意图。
具体实施方式
为使本发明的目的、技术方案和优点更加清楚,下面将结合本发明中的附图,对本发明中的技术方案进行清楚、完整地描述,显然,所描述的实施例是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
下面结合图1-图4描述本发明的复杂纹理花边布的缺陷检测方法。
图1是本发明提供的复杂纹理花边布的缺陷检测方法的流程示意图,如图1所示,该复杂纹理花边布的缺陷检测方法包括以下步骤:
步骤101、获取待检测子图像。
其中,所述待检测子图像中包含花边布图案。
示例地,工业相机采集到包含花边布图案的整幅图像时,由于对整幅图像进行缺陷检测对算力的要求比较高,特别是由于花纹存在周期性,对整幅图像直接检测会造成大量计算冗余。所以,本发明采用滑窗法得到整幅图像的多个待检测子图像;具体地,在花边布传送带上下移动时,从左至右滑窗得到覆盖一整行的待检测子图像;在花边布传送带左右移动时,从上至下滑窗得到覆盖一整列的待检测子图像。
步骤102、确定所述待检测子图像在目标纹理图像中的第一对应位置,将所述第一对应位置处的图像和所述待检测子图像组成待检测图像对。
其中,所述目标纹理图像为花边布图案无缺陷的图像。
示例地,在获取到待检测子图像时,采用Sift特征提取方法对待检测子图像进行特征提取,以确定待检测子图像在目标纹理图像中的第一对应位置,并将第一对应位置处的图像和待检测子图像组成待检测图像对。其中,Sift特征提取方法包括构建尺度空间、检测极值点、过滤极值点和生成特征描述子这四个步骤。
步骤103、将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像。
其中,所述编解码模型是基于多个图像对样本训练得到的;所述编解码模型用于基于所述第一对应位置处的图像对所述待检测子图像进行矫正。
示例地,在得到待检测图像对时,将待检测图像对叠放,形成一个批次,输入至预先训练好的编解码模型中,由编解码模型对待检测图像对中扭曲和畸变的位置进行矫正,得到编解码模型输出的恢复后的待检测子图像,该恢复后的待检测子图像即为恢复扭曲和畸变的待检测子图像。
步骤104、将所述恢复后的待检测子图像和所述目标纹理图像进行比对。
可选地,将所述恢复后的待检测子图像和所述目标纹理图像,从亮度、对比度和结构进行结构相似性评估。
示例地,在得到恢复扭曲和畸变的待检测子图像后,基于结构相似性(StructuralSimilarity,SSIM)的评估方法对目标纹理图像和恢复扭曲和畸变的待检测子图像进行比对,以确定恢复扭曲和畸变的待检测子图像中是否存在缺陷。
具体结构相似性的评估是从亮度、对比度和结构这三个方面进行评估的,若将恢复扭曲和畸变的待检测子图像作为样本x,将目标纹理图像作为样本y,则结构相似性的形式可以采用以下公式(2)来表示:
SSIM(x,y)=l(x,y)α·c(x,y)β·s(x,y)γ (2)
其中,
l表示亮度,c表示对比度,s表示结构,μx表示样本x的平均值,μy表示样本y的平均值,σx表示样本x的方差,σy表示样本y的方差,σxy表示样本x和样本y的协方差,c1、c2和c3为避免除零而添加的常数项,α表示亮度数值的权重,β表示对比度数值的权重,γ表示亮度数值的权重。
具体而言,为了降低评估复杂度,计算时采用滑窗法,每次取一个N×N大小的窗口,计算N×N大小窗口对应的SSIM值,最后取所有窗口的SSIM值的平均值作为全局的SSIM值;其中,N取较小值时可以计算两个图像中对应位置的一个较小邻域之间的结构相似值,计算全图的邻域SSIM即可得到一张待检测子图像,基于SSIM值的大小来确定恢复扭曲和畸变的待检测子图像中是否存在缺陷。
步骤105、在确定所述恢复后的待检测子图像的花边布图案有缺陷时,确定所述恢复后的待检测子图像中的缺陷位置。
示例地,SSIM值接近1的区域为正常区域,SSIM值远离1的区域为缺陷区域,从而可以确定出恢复扭曲和畸变的待检测子图像中的缺陷位置。
本发明提供的一种复杂纹理花边布的缺陷检测方法,将确定的待检测图像对输入至编解码模型中,得到编解码模型输出的恢复后的待检测子图像,然后将恢复后的待检测子图像和花边布图案无缺陷的目标纹理图像进行比对,确定出恢复后的待检测子图像中的缺陷位置,实现了图像缺陷位置的自动检测,从而提高了图像缺陷检测的效率。
可选地,图2是本发明提供的复杂纹理花边布的缺陷检测方法的流程示意图之二,如图2所示,图1中的步骤103之前,该复杂纹理花边布的缺陷检测方法还包括以下步骤:
步骤106、获取多个时间周期的图像样本序列。
其中,所述图像样本序列中至少包括一个时间周期的所述目标纹理图像;所述时间周期为生产流水线中花边布图案循环一次所需的时间。
示例地,通过工业相机采集多个时间周期的图像样本,这些图像样本组成图像样本序列,在图像缺陷检测过程中,需要一系列正常样本图像对程序进行监督,所以,在通过工业相机采集花边布的图像样本时,需要采集包含至少一个时间周期的目标纹理图像,该目标纹理图像为花边布图案无缺陷的图像,将多个目标纹理图像作为正常样本图像,以保障在检测过程中每一个待检测子图像就可以在正常样本图像中找到纹理一致的对应位置。
步骤107、在所述图像样本序列中提取预设尺寸的多个子图像样本。
示例地,在采集到图像样本序列时,在图像样本序列中按照预设尺寸裁剪多个子图像样本,将多个子图像样本作为训练模型需要采用的样本。
步骤108、确定每个所述子图像样本在所述目标纹理图像中的第二对应位置,将每个所述第二对应位置处的图像和对应的所述子图像样本组成多个图像对样本。
可选地,图3是本发明提供的复杂纹理花边布的缺陷检测方法的流程示意图之三,如图3所示,步骤108具体可通过以下步骤实现:
步骤1081、针对每个所述子图像样本,对所述子图像样本构建尺度空间。
示例地,构建尺度空间是使用高斯算子计算不同尺度图像对应的特征图,构建一个多层的高斯金字塔;具体构建方法采用如下公式(3)至公式(5)来表示:
L(x,y,σ)=G(x,y,σ)*I(x,y) (3)
D(x,y,σ)=L(x,y,kσ)-L(x,y,σ) (5)
其中,I表示子图像样本,G表示高斯算子,D表示生成的尺度空间,σ表示尺度空间因子,kσ表示高斯算子的方差,表征图像对应的尺度信息,m和n表示高斯模板的维度,L(x,y,σ)表示变化尺度的高斯函数G(x,y,σ)与子图像样本I(x,y)的卷积,*表示卷积运算。
步骤1082、在所述尺度空间中,确定预设邻域中的极值点。
示例地,预设邻域为3×3×3,则在尺度的高斯金字塔层级中,找到各个3×3×3邻域中的极值点;其中,邻域是指某像素同层周围8个像素以及上下相邻层周围18个,共计26个像素点。
步骤1083、将主曲率比值大于预设阈值的极值点进行过滤。
示例地,主曲率可以由Hessian矩阵获得,假设Hessian矩阵可采用以下公式(6)来表示:
其中,H1(x,y)表示Hessian矩阵,Dxx(x,y)表示对像素点(x,y)的像素值进行两次x轴方向的偏导,Dxy(x,y)表示对像素点(x,y)的像素值先进行x轴方向的偏导,再进行y轴方向的偏导,Dyy(y,y)表示对像素点(x,y)的像素值进行两次y轴方向的偏导。
步骤1084、确定过滤后剩余的每个极值点的特征信息。
其中,特征信息包括位置、尺度和方向这三个方面的信息。
示例地,可以采用k近邻法(k-nearest neighbors)获得一个映射矩阵H2,将子图像样本的四角位置映射到目标纹理图像中;具体地,寻找一个3*3的映射矩阵H2,使得图像对样本中的两个图像(xi,yi)和(x′i,y′i),下述公式(7)的结果最小。
其中,h11、h12、h13、h21、h22、h23、h31、h32和h33分别表示映射矩阵H2中的每个元素,i表示第i个特征点,(xi,yi)表示子图像样本中第i个特征点的坐标,(x′i,y′i)表示目标纹理图像中第i个特征点的坐标。
步骤1085、基于过滤后剩余的每个极值点的特征信息确定所述子图像样本在所述目标纹理图像中的所述第二对应位置。
示例地,在确定子图像样本在目标纹理图像中的第二对应位置时,将每个第二对应位置处的图像和对应的子图像样本组成一个图像对样本;则针对多个子图像样本,就会得到多个图像对样本。
需要说明的是,在训练模型的过程中需要大规模的训练数据集和验证数据集,所以,在图像样本序列中需要随机提取大量预设尺寸的子图像样本,这样就可以得到大规模的图像对样本;本发明通过随机抽取的方式构建大规模无序的图像对样本集合,为后续训练提供训练数据集和验证数据集;例如,本发明生成的图像对样本集合包含5000个训练图像对和1000个测试图像对。
步骤109、基于所述多个图像对样本对深度学习模型进行训练,得到所述编解码模型。
可选地,图4是本发明提供的复杂纹理花边布的缺陷检测方法的流程示意图之四,如图4所示,图2中的步骤109具体可通过以下步骤实现:
步骤1091、将所述多个图像对样本输入至深度学习模型中,得到每个所述图像对样本对应的光流向量场。
其中,深度学习模型包括4层U-Net网络,所述U-Net网络包括一个输入I、一个输出OUT、中间量C1、中间量C2、中间量C3、中间量C4、中间量C5和中间量C6
所述C1=Conv(ω0I,stride=2);
所述C2=Conv(ω1C1,stride=2);
所述C3=Conv(ω2C2,stride=2);
所述C4=Conv(ω3C3,stride=1);
所述C5=Up(Concat(C2,C4));
所述C6=Up(Concat(C1,C5));
所述OUT=Up(Concat(C1,C5));
其中,Conv(I,stride)表示输入为I和步长stride的二维卷积操作,ω表示卷积层的参数,Concat表示两个图像的拼接操作,例如,若输入图像尺寸为c1×h×w以及c2×h×w,则在第一维的角度将两个输入图像拼接为一个(c1+c2)×h×w大小的输出图像,c1表示第一个输入图像的张数,由卷积核的个数确定,c2表示第二个输入图像的张数,由卷积核的个数确定,h表示图像的高度,w表示图像的宽度;Up表示输入图像的上采样操作;例如,采用线性插值法,将输入图像的尺寸放大为2倍。
示例地,在步长为1时,卷积输出尺寸与输入尺寸相同;在步长为2时,卷积输出尺寸是输入尺寸的二分之一;另外,在将多个图像对样本输入至深度学习模型后,由深度学习模型确定图像对样本之间的映射关系,并基于映射关系恢复子图像样本的形态。
其中,图像对样本之间的映射关系为一个尺寸等同于图像对样本尺寸的二层光流向量场g(x,y);例如,图像对样本以及二层光流向量场的尺寸均为2×h×w,二层光流向量场的第一层信息表示横坐标上子图像样本应该位移的距离,第二层信息表示纵坐标上子图像样本应该位移的距离。
需要说明的是,在得到多个图像对样本时,由于模型在训练的时候可能会发生过拟合,所以,本发明引入批归一化(Batch Normalization)方法对卷积层结果进行处理以解决过拟合问题,批归一化的目的是将像素的像素值分布在0到1之间;具体批归一化的流程如表1所示:
表1
步骤1092、将所述光流向量场与对应的所述子图像样本进行叠加,并对叠加结果进行插值,得到恢复后的子图像样本。
示例地,恢复子图像样本的形态是指将二层光流向量场与对应的子图像样本进行叠加,并对未映射到的位置赋以插值得到的值,插值采用双线性插值方法。
例如,在训练模型时,对于得到的多个样本图像对,将多个样本图像对划分为训练集和验证集,将训练集按预设数量个样本图像对为一批输入至深度学习模型,例如预设数量为32,则记为I={(m1,f1),…,(m32,f32)},则得到32个光流向量场v={g1,…,g32},将每个光流向量场与对应的子图像样本进行叠加,并对叠加结果进行插值,得到输出的恢复后的子图像样本,记为y={y1,…,y32}。
步骤1093、基于所述恢复后的子图像样本和所述目标纹理图像构建损失函数。
可选地,采用以下公式(1)构建所述损失函数;
L=Lmse+λLflow (1)
其中,L表示损失函数,λ表示光流向量场梯度损失的权重,其中,λ的取值可以为0.01,Lmse表示所述恢复结果形态损失,Lmse=∑i,j[m′(i,j)-f(i,j)]2;Lflow表示所述光流向量场梯度损失,Lflow=∑i,j[g(i,j,0)2+g(i,j,1)]2;m(i,j)为所述恢复后的子图像样本,f(i,j)为所述目标纹理图像;g(i,j,0)为第一层的光流向量场,g(i,j,1)为第二层的光流向量场。
步骤1094、基于所述损失函数对所述深度学习模型进行优化,直至所述损失函数收敛,得到所述编解码模型。
示例地,需要对深度学习模型进行多个训练,直至最终得到的编解码模型能够正确恢复子图像样本相对于目标纹理图像的扭曲和畸变,其中,正确恢复子图像样本相对于目标纹理图像的扭曲和畸变的评判标准为损失函数,基于损失函数对深度学习模型进行优化,直至损失函数收敛,其中,深度学习模型的优化器可以为Adam优化器,学习率为5×10-4
需要说明的是,全局平均的SSIM值也可以用来评价步骤S109中编解码模型的配准能力,可以在测试时作为判据,或在训练时作为模型的一个损失函数,辅助模型进行训练。
本发明提供的复杂纹理花边布的缺陷检测方法,具备以下有益效果:
1)实现了图像缺陷位置的自动检测,从而提高了图像缺陷检测的效率;
2)本发明的方法为一个自监督深度学习模型,只需学习花边布的纹理形态信息,用目标纹理图像对深度学习模型进行监督,因此对缺陷样本的样式、数量和标注均无要求,能够有效解决实际生产中欠缺缺陷样本和有效标注的问题;
3)本发明设计了一种两阶段高精度配准的复杂纹理花边布的缺陷检测方法:第一阶段,利用Sift特征提取方法提取子图像样本与目标纹理图像的特征点信息,并利用k近邻算法进行匹配,得到子图像样本在目标纹理图像中的对应位置,并形成后续深度学习模型的数据集;第二阶段,采用训练好的编解码模型恢复待检测子图像的相对畸变和扭曲,使得两张图片的形态更为近似。因此,该方法能够高效地对正常样本和待检样本进行比较,具备较高的精度和较好的可解释性。
4)本发明的计算量集中在两个阶段的高精度配准流程中:第一阶段,Sift算子的特征提取和匹配不涉及深度学习计算,计算速度较快;第二阶段,本发明实例采用了轻量级的U-Net作为编解码模型的主干网络,该模型具有特征提取能力强、速度快的优点,模型参数规模不超过200千字节(KB)。因此,本发明能够实现实时检测,具有检测速度快的优点;另外,本发明融合基于特征提取的粗配准算法和基于深度编码-解码器的精配准算法,充分利用花边布的周期性特点,能够实现对复杂纹理花边布的高精度检测。
下面对本发明提供的复杂纹理花边布的缺陷检测装置进行描述,下文描述的复杂纹理花边布的缺陷检测装置与上文描述的复杂纹理花边布的缺陷检测方法可相互对应参照。
图5是本发明提供的复杂纹理花边布的缺陷检测装置的结构示意图,如图5所示,该复杂纹理花边布的缺陷检测装置包括第一获取单元501、第一确定单元502、处理单元503、比对单元504和第二确定单元505;其中:
第一获取单元501,用于获取待检测子图像;其中,所述待检测子图像中包含花边布图案;
第一确定单元502,用于确定所述待检测子图像在目标纹理图像中的第一对应位置,将所述第一对应位置处的图像和所述待检测子图像组成待检测图像对;其中,所述目标纹理图像为花边布图案无缺陷的图像;
处理单元503,用于将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像;
比对单元504,用于将所述恢复后的待检测子图像和所述目标纹理图像进行比对;
第二确定单元505,用于在确定所述恢复后的待检测子图像的花边布图案有缺陷时,确定所述恢复后的待检测子图像中的缺陷位置;
其中,所述编解码模型是基于多个图像对样本训练得到的;所述编解码模型用于基于所述第一对应位置处的图像对所述待检测子图像进行矫正。
本发明提供的一种复杂纹理花边布的缺陷检测装置,将确定的待检测图像对输入至编解码模型中,得到编解码模型输出的恢复后的待检测子图像,然后将恢复后的待检测子图像和花边布图案无缺陷的目标纹理图像进行比对,确定出恢复后的待检测子图像中的缺陷位置,实现了图像缺陷位置的自动检测,从而提高了图像缺陷检测的效率。
基于上述任一实施例,所述比对单元504具体用于:
将所述恢复后的待检测子图像和所述目标纹理图像,从亮度、对比度和结构进行结构相似性评估。
基于上述任一实施例,所述装置还包括第二获取单元、提取单元、第三确定单元和训练单元;
所述第二获取单元,用于获取多个时间周期的图像样本序列;其中,所述图像样本序列中至少包括一个时间周期的所述目标纹理图像;所述时间周期为生产流水线中花边布图案循环一次所需的时间;
所述提取单元,用于在所述图像样本序列中提取预设尺寸的多个子图像样本;
所述第三确定单元,用于确定每个所述子图像样本在所述目标纹理图像中的第二对应位置,将每个所述第二对应位置处的图像和对应的所述子图像样本组成多个图像对样本;
所述训练单元,用于基于所述多个图像对样本对深度学习模型进行训练,得到所述编解码模型。
基于上述任一实施例,所述训练单元具体用于:
将所述多个图像对样本输入至深度学习模型中,得到每个所述图像对样本对应的光流向量场;
将所述光流向量场与对应的所述子图像样本进行叠加,并对叠加结果进行插值,得到恢复后的子图像样本;
基于所述恢复后的子图像样本和所述目标纹理图像构建损失函数;
基于所述损失函数对所述深度学习模型进行优化,直至所述损失函数收敛,得到所述编解码模型。
基于上述任一实施例,所述训练单元还具体用于:
采用以下公式(1)构建所述损失函数;
L=Lmse+λLflow (1)
其中,L表示损失函数,λ表示光流向量场梯度损失的权重,Lmse表示所述恢复结果形态损失,Lmse=∑i,j[m′(i,j)-f(i,j)]2;Lflow表示所述光流向量场梯度损失,Lflow=∑i,j[g(i,j,0)2+g(i,j,1)]2;m′(i,j)为所述恢复后的子图像样本,f(i,j)为所述目标纹理图像;g(i,j,0)为第一层的光流向量场,g(i,j,1)为第二层的光流向量场。
基于上述任一实施例,所述第三确定单元具体用于:
针对每个所述子图像样本,对所述子图像样本构建尺度空间;
在所述尺度空间中,确定预设邻域中的极值点;
将主曲率比值大于预设阈值的极值点进行过滤;
确定过滤后剩余的每个极值点的特征信息;
基于过滤后剩余的每个极值点的特征信息确定所述子图像样本在所述目标纹理图像中的所述第二对应位置。
基于上述任一实施例,所述深度学习模型包括4层U-Net网络,所述U-Net网络包括一个输入I、一个输出OUT、中间量C1、中间量C2、中间量C3、中间量C4、中间量C5和中间量C6
所述C1=Conv(ω0I,stride=2);
所述C2=Conv(ω1C1,stride=2);
所述C3=Conv(ω2C2,stride=2);
所述C4=Conv(ω3C3,stride=1);
所述C5=Up(Concat(C2,C4));
所述C6=Up(Concat(C1,C5));
所述OUT=Up(Concat(C1,C5));
其中,Conv(I,stride)表示输入为I和步长stride的二维卷积操作,ω表示卷积层的参数,Concat表示两个图像的拼接操作,Up表示输入图像的上采样操作。
图6是本发明提供的电子设备的实体结构示意图,如图6所示,该电子设备可以包括:处理器(processor)610、通信接口(Communications Interface)620、存储器(memory)630和通信总线640,其中,处理器610,通信接口620,存储器630通过通信总线640完成相互间的通信。处理器610可以调用存储器630中的逻辑指令,以执行复杂纹理花边布的缺陷检测方法,该方法包括:获取待检测子图像;其中,所述待检测子图像中包含花边布图案;
确定所述待检测子图像在目标纹理图像中的第一对应位置,将所述第一对应位置处的图像和所述待检测子图像组成待检测图像对;其中,所述目标纹理图像为花边布图案无缺陷的图像;
将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像;
将所述恢复后的待检测子图像和所述目标纹理图像进行比对;
在确定所述恢复后的待检测子图像的花边布图案有缺陷时,确定所述恢复后的待检测子图像中的缺陷位置;
其中,所述编解码模型是基于多个图像对样本训练得到的;所述编解码模型用于基于所述第一对应位置处的图像对所述待检测子图像进行矫正。
此外,上述的存储器630中的逻辑指令可以通过软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本发明的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本发明各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、移动硬盘、只读存储器(ROM,Read-Only Memory)、随机存取存储器(RAM,Random Access Memory)、磁碟或者光盘等各种可以存储程序代码的介质。
另一方面,本发明还提供一种计算机程序产品,所述计算机程序产品包括计算机程序,计算机程序可存储在非暂态计算机可读存储介质上,所述计算机程序被处理器执行时,计算机能够执行上述各方法所提供的复杂纹理花边布的缺陷检测方法,该方法包括:获取待检测子图像;其中,所述待检测子图像中包含花边布图案;
确定所述待检测子图像在目标纹理图像中的第一对应位置,将所述第一对应位置处的图像和所述待检测子图像组成待检测图像对;其中,所述目标纹理图像为花边布图案无缺陷的图像;
将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像;
将所述恢复后的待检测子图像和所述目标纹理图像进行比对;
在确定所述恢复后的待检测子图像的花边布图案有缺陷时,确定所述恢复后的待检测子图像中的缺陷位置;
其中,所述编解码模型是基于多个图像对样本训练得到的;所述编解码模型用于基于所述第一对应位置处的图像对所述待检测子图像进行矫正。
又一方面,本发明还提供一种非暂态计算机可读存储介质,其上存储有计算机程序,该计算机程序被处理器执行时实现以执行上述各方法提供的复杂纹理花边布的缺陷检测方法,该方法包括:获取待检测子图像;其中,所述待检测子图像中包含花边布图案;
确定所述待检测子图像在目标纹理图像中的第一对应位置,将所述第一对应位置处的图像和所述待检测子图像组成待检测图像对;其中,所述目标纹理图像为花边布图案无缺陷的图像;
将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像;
将所述恢复后的待检测子图像和所述目标纹理图像进行比对;
在确定所述恢复后的待检测子图像的花边布图案有缺陷时,确定所述恢复后的待检测子图像中的缺陷位置;
其中,所述编解码模型是基于多个图像对样本训练得到的;所述编解码模型用于基于所述第一对应位置处的图像对所述待检测子图像进行矫正。
以上所描述的装置实施例仅仅是示意性的,其中所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。本领域普通技术人员在不付出创造性的劳动的情况下,即可以理解并实施。
通过以上的实施方式的描述,本领域的技术人员可以清楚地了解到各实施方式可借助软件加必需的通用硬件平台的方式来实现,当然也可以通过硬件。基于这样的理解,上述技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来,该计算机软件产品可以存储在计算机可读存储介质中,如ROM/RAM、磁碟、光盘等,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行各个实施例或者实施例的某些部分所述的方法。
最后应说明的是:以上实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明各实施例技术方案的精神和范围。

Claims (7)

1.一种复杂纹理花边布的缺陷检测方法,其特征在于,包括:
获取待检测子图像;其中,所述待检测子图像中包含花边布图案;
确定所述待检测子图像在目标纹理图像中的第一对应位置,将所述第一对应位置处的图像和所述待检测子图像组成待检测图像对;其中,所述目标纹理图像为花边布图案无缺陷的图像;
将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像;
将所述恢复后的待检测子图像和所述目标纹理图像进行比对;
在确定所述恢复后的待检测子图像的花边布图案有缺陷时,确定所述恢复后的待检测子图像中的缺陷位置;
其中,所述编解码模型是基于多个图像对样本训练得到的;所述编解码模型用于基于所述第一对应位置处的图像对所述待检测子图像进行矫正;
在所述将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像之前,所述方法还包括:
获取多个时间周期的图像样本序列;其中,所述图像样本序列中至少包括一个时间周期的所述目标纹理图像;所述时间周期为生产流水线中花边布图案循环一次所需的时间;
在所述图像样本序列中提取预设尺寸的多个子图像样本;
确定每个所述子图像样本在所述目标纹理图像中的第二对应位置,将每个所述第二对应位置处的图像和对应的所述子图像样本组成多个图像对样本;
将所述多个图像对样本输入至深度学习模型中,得到每个所述图像对样本对应的光流向量场;
将所述光流向量场与对应的所述子图像样本进行叠加,并对叠加结果进行插值,得到恢复后的子图像样本;
基于所述恢复后的子图像样本和所述目标纹理图像构建损失函数;
基于所述损失函数对所述深度学习模型进行优化,直至所述损失函数收敛,得到所述编解码模型;
所述基于所述恢复后的子图像样本和所述目标纹理图像构建损失函数,包括:
采用以下公式(1)构建所述损失函数;
(1)
其中,表示损失函数,/>表示光流向量场梯度损失的权重,/>表示恢复结果形态损失,/>;/>所述光流向量场梯度损失,;/>为所述恢复后的子图像样本,/>为所述目标纹理图像;/>为第一层的光流向量场,/>为第二层的光流向量场。
2.根据权利要求1所述的复杂纹理花边布的缺陷检测方法,其特征在于,所述将所述恢复后的待检测子图像和所述目标纹理图像进行比对,包括:
将所述恢复后的待检测子图像和所述目标纹理图像,从亮度、对比度和结构进行结构相似性评估。
3.根据权利要求1所述的复杂纹理花边布的缺陷检测方法,其特征在于,所述确定每个所述子图像样本在所述目标纹理图像中的第二对应位置,包括:
针对每个所述子图像样本,对所述子图像样本构建尺度空间;
在所述尺度空间中,确定预设邻域中的极值点;
将主曲率比值大于预设阈值的极值点进行过滤;
确定过滤后剩余的每个极值点的特征信息;
基于过滤后剩余的每个极值点的特征信息确定所述子图像样本在所述目标纹理图像中的所述第二对应位置。
4.根据权利要求1所述的复杂纹理花边布的缺陷检测方法,其特征在于,所述深度学习模型包括4层U-Net网络,所述U-Net网络包括一个输入、一个输出/>、中间量/>、中间量、中间量/>、中间量/>、中间量/>和中间量/>
所述
所述;
所述;
所述;
所述;
所述;
所述;
其中,表示输入为/>和步长/>的二维卷积操作,/>表示卷积层的参数,/>表示两个图像的拼接操作,/>表示输入图像的上采样操作。
5.一种复杂纹理花边布的缺陷检测装置,其特征在于,包括:
第一获取单元,用于获取待检测子图像;其中,所述待检测子图像中包含花边布图案;
第一确定单元,用于确定所述待检测子图像在目标纹理图像中的第一对应位置,将所述第一对应位置处的图像和所述待检测子图像组成待检测图像对;其中,所述目标纹理图像为花边布图案无缺陷的图像;
处理单元,用于将所述待检测图像对输入至编解码模型中,得到所述编解码模型输出的恢复后的待检测子图像;
比对单元,用于将所述恢复后的待检测子图像和所述目标纹理图像进行比对;
第二确定单元,用于在确定所述恢复后的待检测子图像的花边布图案有缺陷时,确定所述恢复后的待检测子图像中的缺陷位置;
其中,所述编解码模型是基于多个图像对样本训练得到的;所述编解码模型用于基于所述第一对应位置处的图像对所述待检测子图像进行矫正;
所述装置还包括第二获取单元、提取单元、第三确定单元和训练单元;
所述第二获取单元,用于获取多个时间周期的图像样本序列;其中,所述图像样本序列中至少包括一个时间周期的所述目标纹理图像;所述时间周期为生产流水线中花边布图案循环一次所需的时间;
所述提取单元,用于在所述图像样本序列中提取预设尺寸的多个子图像样本;
所述第三确定单元,用于确定每个所述子图像样本在所述目标纹理图像中的第二对应位置,将每个所述第二对应位置处的图像和对应的所述子图像样本组成多个图像对样本;
所述训练单元,用于将所述多个图像对样本输入至深度学习模型中,得到每个所述图像对样本对应的光流向量场;将所述光流向量场与对应的所述子图像样本进行叠加,并对叠加结果进行插值,得到恢复后的子图像样本;基于所述恢复后的子图像样本和所述目标纹理图像构建损失函数;基于所述损失函数对所述深度学习模型进行优化,直至所述损失函数收敛,得到所述编解码模型;
所述训练单元具体用于:
采用以下公式(1)构建所述损失函数;
(1)
其中,表示损失函数,/>表示光流向量场梯度损失的权重,/>表示恢复结果形态损失,/>;/>表示所述光流向量场梯度损失,;/>为所述恢复后的子图像样本,/>为所述目标纹理图像,/>为第一层的光流向量场,/>为第二层的光流向量场。
6.一种非暂态计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至4任一项所述复杂纹理花边布的缺陷检测方法的步骤。
7.一种计算机程序产品,包括计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至4任一项所述复杂纹理花边布的缺陷检测方法的步骤。
CN202210137881.6A 2022-02-15 2022-02-15 复杂纹理花边布的缺陷检测方法及装置 Active CN114565567B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202210137881.6A CN114565567B (zh) 2022-02-15 2022-02-15 复杂纹理花边布的缺陷检测方法及装置

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202210137881.6A CN114565567B (zh) 2022-02-15 2022-02-15 复杂纹理花边布的缺陷检测方法及装置

Publications (2)

Publication Number Publication Date
CN114565567A CN114565567A (zh) 2022-05-31
CN114565567B true CN114565567B (zh) 2024-04-09

Family

ID=81713286

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202210137881.6A Active CN114565567B (zh) 2022-02-15 2022-02-15 复杂纹理花边布的缺陷检测方法及装置

Country Status (1)

Country Link
CN (1) CN114565567B (zh)

Families Citing this family (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN117078677B (zh) * 2023-10-16 2024-01-30 江西天鑫冶金装备技术有限公司 一种用于始极片的缺陷检测方法及系统

Citations (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN107561738A (zh) * 2017-08-30 2018-01-09 湖南理工学院 基于fcn的tft‑lcd表面缺陷快速检测方法
CN110619618A (zh) * 2018-06-04 2019-12-27 杭州海康威视数字技术股份有限公司 一种表面缺陷检测方法、装置及电子设备
CN113838040A (zh) * 2021-09-29 2021-12-24 西安工程大学 一种针对彩色纹理织物缺陷区域的检测方法

Family Cites Families (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
EP3916635B1 (en) * 2020-05-26 2023-05-10 Fujitsu Limited Defect detection method and apparatus

Patent Citations (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN107561738A (zh) * 2017-08-30 2018-01-09 湖南理工学院 基于fcn的tft‑lcd表面缺陷快速检测方法
CN110619618A (zh) * 2018-06-04 2019-12-27 杭州海康威视数字技术股份有限公司 一种表面缺陷检测方法、装置及电子设备
CN113838040A (zh) * 2021-09-29 2021-12-24 西安工程大学 一种针对彩色纹理织物缺陷区域的检测方法

Also Published As

Publication number Publication date
CN114565567A (zh) 2022-05-31

Similar Documents

Publication Publication Date Title
Wu et al. A mutual learning method for salient object detection with intertwined multi-supervision
Talebi et al. Learned perceptual image enhancement
CN110796080B (zh) 一种基于生成对抗网络的多姿态行人图像合成算法
CN109118495B (zh) 一种视网膜血管分割方法和装置
CN106952288B (zh) 基于卷积特征和全局搜索检测的长时遮挡鲁棒跟踪方法
CN104680508B (zh) 卷积神经网络和基于卷积神经网络的目标物体检测方法
CN109902584B (zh) 一种口罩缺陷的识别方法、装置、设备及存储介质
Guo et al. Shadowformer: Global context helps image shadow removal
CN111027464B (zh) 对卷积神经网络和序特征编码联合优化的虹膜识别方法
CN113657560B (zh) 基于节点分类的弱监督图像语义分割方法及系统
Jia et al. Optical flow estimation using learned sparse model
Fan et al. Accurate retinal vessel segmentation via octave convolution neural network
CN110310277B (zh) 基于深度特征和ntv-rpca的织物疵点检测方法
CN106339996B (zh) 一种基于超拉普拉斯先验的图像盲去模糊方法
Rivadeneira et al. Thermal image super-resolution challenge-pbvs 2021
CN110349112A (zh) 一种基于自适应奇异值阈值的两阶段图像去噪方法
CN114565567B (zh) 复杂纹理花边布的缺陷检测方法及装置
Chen et al. Single depth image super-resolution using convolutional neural networks
Zhang et al. Multi-task convnet for blind face inpainting with application to face verification
CN108876776B (zh) 一种分类模型生成方法、眼底图像分类方法及装置
Guo et al. ShadowFormer: Global context helps shadow removal
CN116258632A (zh) 一种基于文本辅助的文本图像超分辨率重建方法
CN111968087B (zh) 一种植物病害区域检测方法
JP7160211B2 (ja) 機械学習モデル、生成装置、コンピュータプログラム
CN116523916B (zh) 产品表面缺陷检测方法、装置、电子设备及存储介质

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination
GR01 Patent grant
GR01 Patent grant