CN114445366B - 基于自注意力网络的长输管道射线影像缺陷智能识别方法 - Google Patents
基于自注意力网络的长输管道射线影像缺陷智能识别方法 Download PDFInfo
- Publication number
- CN114445366B CN114445366B CN202210090694.7A CN202210090694A CN114445366B CN 114445366 B CN114445366 B CN 114445366B CN 202210090694 A CN202210090694 A CN 202210090694A CN 114445366 B CN114445366 B CN 114445366B
- Authority
- CN
- China
- Prior art keywords
- layer
- feature
- attention
- image
- network
- 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
Links
- 230000007547 defect Effects 0.000 title claims abstract description 115
- 238000000034 method Methods 0.000 title claims abstract description 51
- 238000001514 detection method Methods 0.000 claims abstract description 28
- 238000012549 training Methods 0.000 claims abstract description 18
- 230000008569 process Effects 0.000 claims abstract description 8
- 238000004422 calculation algorithm Methods 0.000 claims abstract description 7
- 230000006870 function Effects 0.000 claims description 74
- 239000013598 vector Substances 0.000 claims description 31
- 238000000605 extraction Methods 0.000 claims description 28
- 230000004927 fusion Effects 0.000 claims description 16
- 238000009826 distribution Methods 0.000 claims description 15
- 230000007246 mechanism Effects 0.000 claims description 12
- 238000004364 calculation method Methods 0.000 claims description 11
- 239000011159 matrix material Substances 0.000 claims description 11
- 238000011176 pooling Methods 0.000 claims description 9
- 238000007781 pre-processing Methods 0.000 claims description 9
- 230000004913 activation Effects 0.000 claims description 8
- 238000010586 diagram Methods 0.000 claims description 8
- 230000000694 effects Effects 0.000 claims description 7
- 238000001914 filtration Methods 0.000 claims description 6
- 238000013507 mapping Methods 0.000 claims description 6
- 108010076504 Protein Sorting Signals Proteins 0.000 claims description 3
- 230000001174 ascending effect Effects 0.000 claims description 3
- 238000004519 manufacturing process Methods 0.000 claims description 3
- 238000009659 non-destructive testing Methods 0.000 claims description 3
- 238000010606 normalization Methods 0.000 claims description 3
- 238000013519 translation Methods 0.000 claims description 3
- 230000007306 turnover Effects 0.000 claims description 3
- 238000013461 design Methods 0.000 claims description 2
- 230000008030 elimination Effects 0.000 claims description 2
- 238000003379 elimination reaction Methods 0.000 claims description 2
- 238000002156 mixing Methods 0.000 claims description 2
- 238000013136 deep learning model Methods 0.000 abstract 1
- 238000003466 welding Methods 0.000 description 9
- 238000013527 convolutional neural network Methods 0.000 description 5
- 239000000523 sample Substances 0.000 description 5
- 238000010801 machine learning Methods 0.000 description 3
- 238000012545 processing Methods 0.000 description 3
- 238000013135 deep learning Methods 0.000 description 2
- 230000007613 environmental effect Effects 0.000 description 2
- 238000011156 evaluation Methods 0.000 description 2
- 239000000243 solution Substances 0.000 description 2
- 238000012546 transfer Methods 0.000 description 2
- 238000012935 Averaging Methods 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 230000008859 change Effects 0.000 description 1
- 238000013145 classification model Methods 0.000 description 1
- 238000007796 conventional method Methods 0.000 description 1
- 238000002347 injection Methods 0.000 description 1
- 239000007924 injection Substances 0.000 description 1
- 238000007689 inspection Methods 0.000 description 1
- 230000002452 interceptive effect Effects 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000005457 optimization Methods 0.000 description 1
- 238000003909 pattern recognition Methods 0.000 description 1
- 238000003672 processing method Methods 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 230000011218 segmentation Effects 0.000 description 1
- 230000035945 sensitivity Effects 0.000 description 1
- 238000006467 substitution reaction Methods 0.000 description 1
- 230000001629 suppression Effects 0.000 description 1
- 238000012795 verification Methods 0.000 description 1
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T7/00—Image analysis
- G06T7/0002—Inspection of images, e.g. flaw detection
- G06T7/0004—Industrial image inspection
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/044—Recurrent networks, e.g. Hopfield networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
- G06N3/084—Backpropagation, e.g. using gradient descent
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/30—Subject of image; Context of image processing
- G06T2207/30108—Industrial image inspection
- G06T2207/30152—Solder
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- General Engineering & Computer Science (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Artificial Intelligence (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Biomedical Technology (AREA)
- Life Sciences & Earth Sciences (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Quality & Reliability (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Image Analysis (AREA)
Abstract
本发明提供的一种基于自注意力网络的长输管道射线影像缺陷智能识别方法,具体过程如下:首先将工业底片扫描数字化图像并建立标签数据库。然后基于数据驱动进行深度学习模型建立,在常规目标检测算法中嵌入自注意力模型以增强缺陷特征,并通过颈部网络融合缺陷深层与浅层信息。接着通过联合优化分类与回归损失训练缺陷检测模型。最后基于生成的缺陷检测模型推理无标记图像获得缺陷的位置信息以及类别信息。本发明提供的方法大幅度提升了长输管道射线影像智能识别能力。
Description
技术领域
本发明涉及管道焊缝缺陷检测技术领域,尤其涉及基于自注意力网络的长输管道射线影像缺陷智能识别方法。
背景技术
目前石化工业迅速发展,长输管道在运输能源中扮演着关键角色,因此维护管道的安全至关重要。在众多管道故障当中由于焊缝质量不合格而引起的泄露事故不在少数,主要原因在于焊接过程中产生了许多危害性较高的缺陷,这些缺陷违背及时识别将严重威胁长时间工作在恶劣环境下管道的安全。因此为了保证焊接工程质量,避免因未能及时发现焊接严重缺陷而造成管道泄漏,给环境经济和安全造成巨大损失,必须严格按照焊接标准对管道焊缝区域进行缺陷检查。
由于人工缺陷检测的低效性与主观性,基于X射线的自动缺陷识别系统被广泛研究。在目前的方法中,大部分的研究内容集中在两个方向,其一是基于传统的图像处理和机器学习方法,其二是基于深度学习算法实现端到端的焊缝缺陷检测。在传统的方法中,基于图像处理的方法通常用于提高X射线图像的质量,并从图像中分割出感兴趣的区域,基于机器学习方法通常用于模式识别。基于滤波的方法为X射线图像处理常用的手段,通常采用像素阈值分割等形式实现缺陷图像二值化,然后依据不同类别缺陷的特征不同基于机器学习可解释分类模型实现最终的缺陷检测。目前该方法的缺点有如下几点:缺陷检测流程繁琐,特征提取需要依赖大量专业人员经验,传统方法对复杂缺陷特征学习能力较弱,无法准确识别长输管线复杂的缺陷影响。
目前深度学习广泛应用到工业领域,大量的端到端目标识别算法在工业检测方面取得了良好的进展,但是目前长输管道焊接缺陷检测存在以下难点:其一、图像背景复杂,某些背景特征与缺陷及其相似,传统的卷积网络结构对齐拟合能力较弱。其二、微小缺陷影像特征较弱,常规网络结构无法对其精准识别。
目前注意机制已成为CNN模型不可缺少的一部分,在抑制复杂背景和突出物体特征方面表现出良好的效果,它可以通过融合通道特征和空间特征来实现对噪声特征的抑制和对缺陷特征的突出。
发明内容
针对目前焊缝缺陷检测精度较低的技术问题,本发明提供了一种基于自注意力网络的长输管道射线影像缺陷智能识别方法。
具体技术方案如下:
基于自注意力网络的长输管道射线影像缺陷智能识别方法,包括以下步骤:
步骤1:基于NEU-ZD-2工业底片扫描仪对工业底片进行数字化获得焊缝缺陷数字化图像后,基于无损检测工程师进行缺陷样本标注,制作数据集;
步骤2:图像预处理,焊接区域的粗糙度,原始图像包含了大量的干扰信号,这显著地影响了对缺陷的识别,采用改进的均值滤波方法去除原始X射线图像所包含的大量干扰信号,完成去除噪声预处理;
步骤3:设计主干网络实现缺陷特征提取,将制作的焊缝缺陷图像与其相对应的缺陷标签的输入到网络中,提取纹理和语义特征;
步骤4:设计改进的自注意力网络层增强主干网络提取到的缺陷深层特征;自注意力层用于计算特征中不同位置之间的权重,从而达到更新特征的效果,通过连接层进行特征的映射,得到增强后的特征。
步骤5:对多级特征进行特征融合,实现深层浅层特征增强;
步骤6:模型预测机制:在预测头部对颈部网络所提出的三个特征图进行推理预测,使用三种不同尺度的特征图来预测不同大小的对象;在预测过程中,输入图像被分成几个在回归时通过预先在特征图上设置Anchors,检测头中的每个网格分配三种类型的锚:
步骤7:分类及回归损失函数设计:损失函数用于评估预测值与真实值之间的差值,将训练集分批次送入网络结构,通过最小化公式损失函数训练检测器从而优化参数,该损失函数包括两部分,分别是分类损失函数与回归损失函数。
步骤1具体包括如下步骤:
步骤1-1,标注缺陷数据集,按照矩形框的形式给出缺陷的像素坐标;对于给定训练集图像{XL,YL},任意图像xi中的缺陷实例集可以表示为每张图像的实例标签记作
步骤1-2:进行数据增强,扩充训练样本来增加训练样本的多样性,提高模型泛化能力,本文采取平移,翻折,尺度缩放,图像拼接方式并对标签采取同样操作从而大量扩充了数据集,获得{XL,YL}DA。
步骤2所述的图像预处理具体包括以下步骤:
步骤2-1:去除该平均值是为了消除探头的背景信号;由于环境的变化以及X射线图像扫描仪自身参数的干扰,扫描出的图像存在干扰影像,对原始的焊缝缺陷图像中的每个通道进行均值滤波,消除了由于扫描仪硬件生成的干扰。
步骤2-2:为了避免图像中一些振幅极高的信号支配微弱缺陷信号的可能性,有必要限制图像的振幅,因此设置幅值截断和均值滤波器:首先将图像矩阵中的像素的最大值被限制为低于一个阈值;然后应用一个中值滤波器到每个通道;设计的改进中值滤波器算法实现如下:假设信号序列为x(j),j∈n,定义一个奇数长度为L=2N+1的滑动窗口,其中N是一个正整数;对于每个整数i,L个样本,即x(i-N)x(i),Λx(i+N),按升序排列,然后将中值定义为中值滤波器的输出;
步骤2-3:梯度分数计算:图像的梯度被写成X方向和Y方向上的梯度之和,如下式所示:这里f(i,j)表示位置(i,j)处的像素值,xnum是x轴方向上的数据点总数,NX是用于计算每个位置上的梯度分数的数据点数;在y轴方向上的梯度分数的计算过程与上述计算过程相似。
步骤3所述的设计主干网络实现缺陷特征提取,具体包括以下步骤:
步骤3-1:输入图像到特征提取网络中,提取纹理和语义特征;输入模型图像的大小必须是32的倍数,这里输入图像参数选取640*640*3;
步骤3-2:逐层特征提取:将输入图像依次送入主干网络各层,提取缺陷特征;
步骤3-3:多层特征提取:按照上述机制重复多次进行深层特征提取,得到多阶段特征图。
步骤4所述的设计改进的自注意力网络层增强主干网络提取到的缺陷深层特征,具体包括以下步骤:
步骤4-1:确定输入深层特征:在这里,将主干网络层最后一层高级特征图X5(H=20,W=20,C=1024),
其中,H为图像特征的高度,W为图像特征的宽度,C为通道数,维度图像拉成一维向量,记作深层特征序列向量x=(x1,Λ,xn),n=20*20*1024,其中n代表向量的维度;
步骤4-2:对深层特征向量位置编码,保证特征向量空间位置信息,具体过程如下:对输入向量注入关于序列的相对或绝对位置的信息序列中的标记,即在输入向量中嵌入位置编码,位置编具有与输入相同的维数模型,这里使用不同频率的正弦函数实现输入位置端编码:
其中pos为位置,i为输入维度,编码后向量维度是dmodel;
步骤4-3,确定编码器结构:该编码器由六层结构组成,每一层集包含两个子层,其一为多头注意力网络层,其二为前馈网络结构层;每个子层周使用残差连接,并进行归一化,每个子层的输出为:sublayer=LayerNorm(x+Sublayer(x)),
其中x为输入特征sublayer为输出特征,LayerNorm为当前阶段的网络激活层,Sublayer(x)是由子层本身实现的函数;模型中的所有子层以及嵌入层都会产生dmodel=512维的输出;
步骤4-4,确定注意力函数层:自注意力机制就是要通过权重矩阵来自发地找到不同特征之间的关系;该函数可以描述为将查询和一组键值对映射到输出,其中查询、键、值和输出都是向量;输出作为值的加权和计算,其中分配给每个值的权重由查询与相应键的兼容性函数计算;
首先按照高斯分布初始化参数值Q,K,V权值,基于固定方差的参数进行高斯分布初始化:Q,K,V~N(μ,σ),这里μ=0,σ=1从均值为0,方差为1的高斯分布,并将所得向量作为初始权值,并通过使用
Softmax函数来获得值的权重,因此该输出权重矩阵为:
其中,为输出权重矩阵,Q,K,V分别为初始化后的全连接层分健参数别代表查询量,健参数与值参数,softmax为权重激活函数,dk为健参数K的向量维度;
步骤4-5,确定多层注意力函数:多头注意力模型共同关注来自不同位置的不同表示子空间的信息,其中注意力头可以描述为:
其中投影参数
MultiHead(Q,K,V)=Concat(head1,Λ,headh)wO,在这里使用使用h=8平行注意层;
步骤4-4:确定前馈网络层:除注意力子层外,编码器和解码器中的每个层都包含一个完全连接的前馈网络FFN(x),具体包括两个线性变换,中间采用ReLU激活,其中前馈网络层公式如下:
FFN(x)=max(0,xW1+b1)W2+b2;
其中,x为输入特征,W1,b1和W2,b2分别为前馈网络第一层和第二层权重与偏置,max函数为比较最值函数;
步骤4-5:确定解码层:按照上述编码结构对注意力层进行逐层解码,即进行逆运算操作。
步骤5所述的对多级特征进行特征融合,实现深层浅层特征增强,具体包括以下步骤:
步骤5-1:基于改进的SPP层实现顶层注意力特征图池化操作,对于选择的不同大小的区域对应到卷积之后的特征图上,得到的也是大小不一致的特征图区域,对于每个区域,通过三种划分方式进行池化:按照如下公式所示:
Xpool=Concat(pool5*5(Xattention),pool7*7(Xattention),pool13*13(Xattention))
其中,Xpool为输出池化层的特征,Concat为张量拼接函数,pool5*5,pool7*7和pool13*13分别代表卷积核分别为5,7,13的池化运算,Xattention为输入注意力特征图。
步骤5-2:特征金字塔自上而下融合缺陷特征:记X1和X2表示来自特征提取网络的不同阶段的全局特征图,并具有不同的分辨率,通过整合多尺度全局特征和多尺度局部区域特征,可以增强网络表达焊缝缺陷特征的能力,利用通道上SPP块的输入特征图,20×20×1024特征图作为对象检测的输出,全局特征融合和局部特征融合连续连接,拼接公式如下:
其中,Xi表示特征提取网络中不同阶段的特征图;表示每个源特征映射在连接在一起之前的转换函数;Concat表示沿着通道连接的特征映射即张量拼接运算,代表自上而下融合时消除频率混叠函数;
步骤5-3:特征金字塔PAN自下而上融合缺陷特征:
其中,Concat为张量拼接运算,代表自上而下融合时消除频率混叠函数,F2为最终输出特征表示。
步骤7具体包括如下步骤:
步骤7-1:损失函数用于评估预测值与真实值之间的差值,将训练集分批次送入网络结构,通过最小化公式损失函数训练检测器从而优化参数Θ,该损失函数包括两部分,分别是分类损失函数与回归损失函数,按照如下公式所示:
其中FL(·)为Focal分类损失函数,CIoU(·)为边界框位置回归损失函数,和分别代表网络预测图像的分类概率分布和位置回归概率分布, 和分别代表图像yi中真实的类别标签和位置标签;
步骤7-2:权值更新:通过反向传播更新权重的目的是为了最小化损失,将损失作为权值参数的函数,CNN需要计算损失相对于每个权值参数的偏导数,然后使用随机梯度下降(SGD)方法在梯度下降最快的方向迭代更新权值,直到满足参数停止更新的条件。
与现有技术相比,本发明具有如下有益技术效果:
本发明是一种新的基于CNN与自注意力的缺陷检测网络来识别长输管道缺陷影像,基于本发明提供一种基于自注意力机制下的长输管道缺陷影像识别方法。基于通用主干网络Darknet53网络添加自注意力层,并设计了多通道跨阶段层融合主干网络。利用卷积神经网络构建自注意力导向的特征提取模块,使得缺陷的检出率有了明显的提升。该网络采用先进的主干网络模块,通过跨尺度连接在主干最后三层中选择信息丰富的多尺度特征。该操作可以大大提高X射线中小缺陷识别影响的特征表示能力。所提出的网络结构在缺陷分类和检测性能方面具有较高的精度。此外由于本发明缺陷识别速度快,在实际的常熟管道缺陷影像识别任务中具有很大的应用潜力。
附图说明
图1为本发明实施例提供的一种基于自注意力网络的长输管道缺陷影像智能识别方法检测方法总结构图;
图2为本发明实施例提供的训练验证评价指标图;
图3为本发明实施例提供的检测结果图。
具体实施方式
下面结合附图和具体实施例对本发明进行详细说明,但本发明的保护范围不受附图和具体实施例所限。
实施例1:
一种基于自注意力网络的长输管道缺陷影像智能识别方法检测方法,该方法主要包括三大部分:数据预处理部分,特征提取与融合部分,预测部分,整体结构图如图1所示。首先将原始对原始的焊缝图像进优化处理,然后送入注入注意力机制的的特征提取主干网络,然后基于增强特征进行最终预测,具体步骤如下:
步骤1:基于NEU-ZD-2工业底片扫描仪对工业底片进行数字化获得焊缝缺陷数字化图像后,基于无损检测工程师进行缺陷样本标注,制作数据集,获取焊缝缺陷图像。
步骤1-1,标注缺陷数据集,按照矩形框的形式给出缺陷的像素坐标。对于给定训练集图像{XL,YL},任意图像xi中的缺陷实例集可以表示为每张图像的实例标签记作其中标记了六种缺陷类别分别为圆形缺陷,条形缺陷,未熔合缺陷,未焊透缺陷,烧穿缺陷,裂纹缺陷,标签分别记作r,l,lf,ip,bt,c。步骤1-2:进行数据增强,扩充训练样本来增加训练样本的多样性,提高模型泛化能力,本文采取平移,翻折,尺度缩放,图像拼接方式并对标签采取同样操作从而大量扩充了数据集,获得{XL,YL}DA。
步骤2:图像预处理:进行均置滤波去除图像噪声。焊接区域的粗糙度,原始图像包含了大量的干扰信号,这显著地影响了对缺陷的识别。因此,需要先对原始图像进行过滤。该算法的步骤如下所示:
步骤2-1:从图像中去除每个通道的平均值。去除该平均值是为了消除探头的背景信号。由于环境的变化以及X射线图像扫描仪自身参数的干扰,扫描出的图像存在干扰影像。
步骤2-2:幅值截断和均值滤波器。为了避免图像中一些振幅极高的信号支配微弱缺陷信号的可能性,有必要限制图像的振幅。因此,图像矩阵中的像素的最大值被限制为低于一个阈值。然后应用一个中值滤波器到每个通道。中值滤波器算法实现如下:假设信号序列为x(j),j∈n,定义一个奇数长度为L=2N+1的滑动窗口,其中N是一个正整数。对于每个整数i,L个样本,即x(i-N)x(i),Λx(i+N),按升序排列,然后将中值定义为中值滤波器的输出。
步骤2-3:梯度分数计算。图像的梯度被写成X方向和Y方向上的梯度之和,如下式所示:这里f(i,j)表示位置(i,j)处的像素值,xnum是x轴方向上的数据点总数,NX是用于计算每个位置上的梯度分数的数据点数。在y轴方向上的梯度分数的计算过程与上述计算过程相似。其中四种标注缺陷样本图如图2所示,左侧为原始图像,右侧为图像处理后的效果。
步骤3:设计主干网络实现缺陷特征提取。将制作的焊缝缺陷图像与其相对应的缺陷标签的输入到网络中,提取纹理和语义特征。
步骤3-1:输入图像到特征提取网络中,提取纹理和语义特征。输入模型图像的大小必须是32的倍数,这里输入图像参数选取640*640*3,其中
步骤3-2:逐层特征提取,将输入图像依次送入主干网络各层,提取缺陷特征。对于每个主干网络密集层(Block)层包含多个卷积层,对于一组卷积核结构Con(k,s,p,n),其中k表示卷积核的大小,s表示步长,p表示零填充,n表示卷积核的数量。因此,特征提取过程可以描述为:
其中,是i层输入的特征图,是第i-1卷积层输出的特征图。
特征图尺寸为u*u,并且其满足
在卷积层和激活单元层之间使用批归一化(BN)层来加快网络训练,降低对网络初始化的敏感性,BN层计算按照如下所示:
其中,是主干网络结构中第i阶段层的卷积层特征图。
LeakyRelu可以解决传统的激活函数(Relu)当输入接近零或为负时,函数的梯度变为零,网络将无法执行反向传播的不足,可获得阶段特征图Xi
步骤3-3:多层特征提取:
按照上述机制重复多次进行深层特征提取,得到多阶段特征图,其具体Block结构满足下表所示:
阶段 | 卷积核参数 | 特征图尺寸 |
Block 1 | [Conv(1,1,0,32)+Conv(3,1,1,64)]×1 | (320,320,64) |
Block 2 | [Conv(1,1,0,64)+Conv(3,1,1,128)]×2 | (160,160,128) |
Block 3 | [Conv(1,1,0,128)+Conv(3,1,1,256)]×8 | (80,80,256) |
Block 4 | [Conv(1,1,0,256)+Conv(3,1,1,512)]×16 | (40,40,512) |
Block 5 | [Conv(1,1,0,512)+Conv(3,1,1,1024)]×4 | (20,20,1024) |
步骤4:设计改进的自注意力网络层增强主干网络提取到的缺陷深层特征。自注意力层用于计算特征中不同位置之间的权重,从而达到更新特征的效果,通过连接层进行特征的映射,得到一个新的特征,具体步骤如下所示:
步骤4-1:确定输入深层特征:在这里,将主干网络层最后一层高级特征图X5(H=20,W=20,C=1024)维度图像拉成一维向量,记作深层特征序列向量x=(x1,Λ,xn),n=20*20*1024。
步骤4-2:对深层特征向量位置编码,保证特征向量空间位置信息。具体过程如下:对输入向量注入关于序列的相对或绝对位置的信息序列中的标记,即在输入向量中嵌入位置编码,位置编具有与输入相同的维数模型,这里使用不同频率的正弦函数实现输入位置端编码:
其中pos是位置,i是维度,编码后向量维度是dmodel。
步骤4-3,确定编码器结构:该编码器由六层结构组成,每一层集包含两个子层,其一为多头注意力网络层,其二为前馈网络结构层。每个子层周使用残差连接,并进行归一化。每个子层的输出为:sublayer=LayerNorm(x+Sublayer(x)),其中Sublayer(x)是由子层本身实现的函数。模型中的所有子层以及嵌入层都会产生dmodel=512维的输出。
步骤4-4,确定注意力函数层:自注意力机制就是要通过权重矩阵来自发地找到不同特征之间的关系。该函数可以描述为将查询和一组键值对映射到输出,其中查询、键、值和输出都是向量。输出作为值的加权和计算,其中分配给每个值的权重由查询与相应键的兼容性函数计算。
首先按照高斯分布初始化参数值Q,K,V权值,基于固定方差的参数进行高斯分布初始化:Q,K,V~N(μ,σ),这里μ=0,σ=1从均值为0,方差为1的高斯分布,并将所得向量作为初始权值,并通过使用Softmax函数来获得值的权重。因此该输出矩阵为:
步骤4-5,确定多层注意力函数:多头注意力模型共同关注来自不同位置的不同表示子空间的信息。其中注意力头可以描述为:
其中投影参数
MultiHead(Q,K,V)=Concat(head1,Λ,headh)wO,在这里使用使用h=8平行注意层。
步骤4-4:确定前馈网络层:除了注意力子层外,编码器和解码器中的每个层都包含一个完全连接的前馈网络FFN(x),具体包括两个线性变换,中采用ReLU激活,其中前馈网络层公式如下:FFN(x)=max(0,xW1+b1)W2+b2。
步骤4-5:确定解码层:按照上述编码结构对注意力层进行逐层解码,即进行逆运算操作。
步骤5:对多级特征进行特征融合,实现深层浅层特征增强。
步骤5-1:基于改进的SPP层实现顶层注意力特征图池化操作。对于选择的不同大小的区域对应到卷积之后的特征图上,得到的也是大小不一致的特征图区域,对于每个区域,通过三种划分方式进行池化:按照如下公式所示:
Xpool=Concat(pool5*5(Xattention),pool7*7(Xattention),pool13*13(Xattention))
步骤5-2:特征金字塔自上而下融合缺陷特征:记X1和X2表示来自特征提取网络的不同阶段的全局特征图,并具有不同的分辨率。通过整合多尺度全局特征和多尺度局部区域特征,可以增强网络表达焊缝缺陷特征的能力。利用通道上SPP块的输入特征图,20×20×1024特征图作为对象检测的输出。全局特征融合和局部特征融合连续连接,拼接公式如下:
步骤5-3:特征金字塔PAN自上而下融合缺陷特征:
Xi表示特征提取网络中不同阶段的特征图。表示每个源特征映射在连接在一起之前的转换函数。Concat表示沿着通道连接的特征映射。f1,f2表示多尺度全局特征融合后的输出特征图,
步骤6:模型预测机制:在预测头部对颈部网络所提出的三个特征图进行推理预测,使用三种不同尺度的特征图来预测不同大小的对象。在预测过程中,输入图像被分成几个在回归时通过预先在特征图上设置Anchors,检测头中的每个网格分配三种类型的锚。
步骤7:损失函数与权重更新:
步骤7-1:损失函数用于评估预测值与真实值之间的差值,将训练集分批次送入网络结构,通过最小化公式损失函数训练检测器从而优化参数Θ,该损失函数包括两部分,分别是分类损失函数与回归损失函数,如公式(2)所示:
其中FL(·)为Focal分类损失函数,CIoU(·)为边界框位置回归损失函数,和分别代表网络预测图像的分类概率分布和位置回归概率分布, 和分别代表图像yi中真实的类别标签和位置标签。其训练损失与评价曲线如图2所示:
步骤7-2:权值更新:通过反向传播更新权重的目的是为了最小化损失,将损失作为权值参数的函数,CNN需要计算损失相对于每个权值参数的偏导数,然后使用随机梯度下降(SGD)方法在梯度下降最快的方向迭代更新权值,直到满足参数停止更新的条件。损失不下降时停止更新,获取缺陷检测模型,最终预测结构图如图3所示,虚线框左侧为现有公开的YOLO-v5原始的检测器效果,右侧为本发明提高的模型,可以看出缺陷识别能力较高,漏检与误检均相对较少。
本实施例提供一种基于自注意力网络的长输管道缺陷影像智能识别方法检测方法,该方法主要包括三大部分:数据预处理部分,特征提取与融合部分,预测部分。通过实际缺陷检测可以看出,本发明可以很好的解决当前缺陷识别精度较低的问题。最后应说明的是:以上各实施例仅用以说明本发明的技术方案,而非对其限制;尽管参照前述各实施例对本发明进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分或者全部技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本发明权利要求所限定的范围。
Claims (7)
1.基于自注意力网络的长输管道射线影像缺陷智能识别方法,其特征在于,包括以下步骤:
步骤1:基于NEU-ZD-2工业底片扫描仪对工业底片进行数字化获得焊缝缺陷数字化图像后,基于无损检测工程师进行缺陷样本标注,制作数据集;
步骤2:图像预处理,采用改进的均值滤波方法去除原始X射线图像所包含的大量干扰信号,完成去除噪声预处理;
步骤3:设计主干网络实现缺陷特征提取,将制作的焊缝缺陷图像与其相对应的缺陷标签的输入到网络中,提取纹理和语义特征;
步骤4:设计改进的自注意力网络层增强主干网络提取到的缺陷深层特征;自注意力层用于计算特征中不同位置之间的权重,从而达到更新特征的效果,通过连接层进行特征的映射,得到增强后的特征;
步骤5:对多级特征进行特征融合,实现深层浅层特征增强;
步骤6:模型预测机制:在预测头部对颈部网络所提出的三个特征图进行推理预测,使用三种不同尺度的特征图来预测不同大小的对象;在预测过程中,输入图像被分成几个在回归时通过预先在特征图上设置Anchors,检测头中的每个网格分配三种类型的锚:
步骤7:分类及回归损失函数设计:损失函数用于评估预测值与真实值之间的差值,将训练集分批次送入网络结构,通过最小化公式损失函数训练检测器从而优化参数,该损失函数包括两部分,分别是分类损失函数与回归损失函数。
2.根据权利要求1所述的基于自注意力网络的长输管道射线影像缺陷智能识别方法,其特征在于:步骤1具体包括如下步骤:
步骤1-1,标注缺陷数据集,按照矩形框的形式给出缺陷的像素坐标;对于给定训练集图像{XL,YL},任意图像xi中的缺陷实例集可以表示为每张图像的实例标签记作
步骤1-2:进行数据增强,扩充训练样本来增加训练样本的多样性,提高模型泛化能力,本文采取平移,翻折,尺度缩放,图像拼接方式并对标签采取同样操作从而大量扩充了数据集,获得{XL,YL}DA。
3.根据权利要求1所述的基于自注意力网络的长输管道射线影像缺陷智能识别方法,其特征在于:步骤2所述的图像预处理具体包括以下步骤:
步骤2-1:对原始的焊缝缺陷图像中的每个通道进行均值滤波,消除了由于扫描仪硬件生成的干扰;
步骤2-2:设置幅值截断和均值滤波器:首先将图像矩阵中的像素的最大值被限制为低于一个阈值;然后应用一个中值滤波器到每个通道;设计的改进中值滤波器算法实现如下:假设信号序列为x(j),j∈n,定义一个奇数长度为L=2N+1的滑动窗口,其中N是一个正整数;对于每个整数i,L个样本,即x(i-N)x(i),Λx(i+N),按升序排列,然后将中值定义为中值滤波器的输出;
步骤2-3:梯度分数计算:图像的梯度被写成X方向和Y方向上的梯度之和,如下式所示:这里f(i,j)表示位置(i,j)处的像素值,xnum是x轴方向上的数据点总数,NX是用于计算每个位置上的梯度分数的数据点数;在y轴方向上的梯度分数的计算过程与上述计算过程相似。
4.根据权利要求1所述的基于自注意力网络的长输管道射线影像缺陷智能识别方法,其特征在于:步骤3所述的设计主干网络实现缺陷特征提取,具体包括以下步骤:
步骤3-1:输入图像到特征提取网络中,提取纹理和语义特征;输入模型图像的大小必须是32的倍数,这里输入图像参数选取640*640*3;
步骤3-2:逐层特征提取:将输入图像依次送入主干网络各层,提取缺陷特征;
步骤3-3:多层特征提取:按照上述机制重复多次进行深层特征提取,得到多阶段特征图。
5.根据权利要求1所述的基于自注意力网络的长输管道射线影像缺陷智能识别方法,其特征在于:步骤4所述的设计改进的自注意力网络层增强主干网络提取到的缺陷深层特征,具体包括以下步骤:
步骤4-1:确定输入深层特征:在这里,将主干网络层最后一层高级特征图X5(H=20,W=20,C=1024),
其中,H为图像特征的高度,W为图像特征的宽度,C为通道数,维度图像拉成一维向量,记作深层特征序列向量x=(x1,Λ,xn),n=20*20*1024,其中n代表向量的维度;
步骤4-2:对深层特征向量位置编码,保证特征向量空间位置信息,具体过程如下:对输入向量注入关于序列的相对或绝对位置的信息序列中的标记,即在输入向量中嵌入位置编码,位置编具有与输入相同的维数模型,这里使用不同频率的正弦函数实现输入位置端编码:
其中pos为位置,i为输入维度,编码后向量维度是dmodel;
步骤4-3,确定编码器结构:该编码器由六层结构组成,每一层集包含两个子层,其一为多头注意力网络层,其二为前馈网络结构层;每个子层周使用残差连接,并进行归一化,每个子层的输出为:sublayer=LayerNorm(x+Sublayer(x)),
其中x为输入特征sublayer为输出特征,LayerNorm为当前阶段的网络激活层,Sublayer(x)是由子层本身实现的函数;模型中的所有子层以及嵌入层都会产生dmodel=512维的输出;
步骤4-4,确定注意力函数层:自注意力机制就是要通过权重矩阵来自发地找到不同特征之间的关系;该函数可以描述为将查询和一组键值对映射到输出,其中查询、键、值和输出都是向量;输出作为值的加权和计算,其中分配给每个值的权重由查询与相应键的兼容性函数计算;
首先按照高斯分布初始化参数值Q,K,V权值,基于固定方差的参数进行高斯分布初始化:Q,K,V~N(μ,σ),这里μ=0,σ=1从均值为0,方差为1的高斯分布,并将所得向量作为初始权值,并通过使用
Softmax函数来获得值的权重,因此该输出权重矩阵为:
其中,为输出权重矩阵,Q,K,V分别为初始化后的全连接层分健参数别代表查询量,健参数与值参数,softmax为权重激活函数,dk为健参数K的向量维度;
步骤4-5,确定多层注意力函数:多头注意力模型共同关注来自不同位置的不同表示子空间的信息,其中注意力头可以描述为:
head1=Attention(QWi Q,QWi K,VWi V),
其中投影参数MultiHead(Q,K,V)=Concat(head1,Λ,headh)wO,在这里使用使用h=8平行注意层;
步骤4-4:确定前馈网络层:除注意力子层外,编码器和解码器中的每个层都包含一个完全连接的前馈网络FFN(x),具体包括两个线性变换,中间采用ReLU激活,其中前馈网络层公式如下:
FFN(x)=max(0,xW1+b1)W2+b2;
其中,x为输入特征,W1,b1和W2,b2分别为前馈网络第一层和第二层权重与偏置,max函数为比较最值函数;
步骤4-5:确定解码层:按照上述编码结构对注意力层进行逐层解码,即进行逆运算操作。
6.根据权利要求1所述的基于自注意力网络的长输管道射线影像缺陷智能识别方法,其特征在于:步骤5所述的对多级特征进行特征融合,实现深层浅层特征增强,具体包括以下步骤:
步骤5-1:基于改进的SPP层实现顶层注意力特征图池化操作,对于选择的不同大小的区域对应到卷积之后的特征图上,得到的也是大小不一致的特征图区域,对于每个区域,通过三种划分方式进行池化:按照如下公式所示:
Xpool=Concat(pool5*5(Xattention),pool7*7(Xattention),pool13*13(Xattention))
其中,Xpool为输出池化层的特征,Concat为张量拼接函数,pool5*5,pool7*7和pool13*13分别代表卷积核分别为5,7,13的池化运算,Xattention为输入注意力特征图;
步骤5-2:特征金字塔自上而下融合缺陷特征:记X1和X2表示来自特征提取网络的不同阶段的全局特征图,并具有不同的分辨率,通过整合多尺度全局特征和多尺度局部区域特征,可以增强网络表达焊缝缺陷特征的能力,利用通道上SPP块的输入特征图,20×20×1024特征图作为对象检测的输出,全局特征融合和局部特征融合连续连接,拼接公式如下:
其中,Xi表示特征提取网络中不同阶段的特征图;F1自上而下特征融合时产生的特征图集合;Concat表示沿着通道连接的特征映射即张量拼接运算,代表自上而下融合时消除频率混叠函数;
步骤5-3:特征金字塔PAN自下而上融合缺陷特征:
其中,Concat为张量拼接运算,代表自上而下融合时消除频率混叠函数,F2为最终输出特征表示。
7.根据权利要求1所述的基于自注意力网络的长输管道射线影像缺陷智能识别方法,其特征在于,步骤7具体包括如下步骤:
步骤7-1:损失函数用于评估预测值与真实值之间的差值,将训练集分批次送入网络结构,通过最小化公式损失函数训练检测器从而优化参数Θ,该损失函数包括两部分,分别是分类损失函数与回归损失函数,按照如下公式所示:
其中FL(·)为Focal分类损失函数,CIoU(·)为边界框位置回归损失函数,和分别代表网络预测图像的分类概率分布和位置回归概率分布, 和分别代表图像yi中真实的类别标签和位置标签;
步骤7-2:权值更新:通过反向传播更新权重的目的是为了最小化损失,将损失作为权值参数的函数,CNN需要计算损失相对于每个权值参数的偏导数,然后使用随机梯度下降(SGD)方法在梯度下降最快的方向迭代更新权值,直到满足参数停止更新的条件。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210090694.7A CN114445366B (zh) | 2022-01-26 | 2022-01-26 | 基于自注意力网络的长输管道射线影像缺陷智能识别方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210090694.7A CN114445366B (zh) | 2022-01-26 | 2022-01-26 | 基于自注意力网络的长输管道射线影像缺陷智能识别方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN114445366A CN114445366A (zh) | 2022-05-06 |
CN114445366B true CN114445366B (zh) | 2024-08-09 |
Family
ID=81370176
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210090694.7A Active CN114445366B (zh) | 2022-01-26 | 2022-01-26 | 基于自注意力网络的长输管道射线影像缺陷智能识别方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN114445366B (zh) |
Families Citing this family (12)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN114491083B (zh) * | 2022-04-01 | 2022-07-15 | 江苏智云天工科技有限公司 | 工业检测中的知识图库构建方法、知识图库构建装置 |
CN114943251B (zh) * | 2022-05-20 | 2023-05-02 | 电子科技大学 | 一种基于融合注意力机制的无人机目标识别方法 |
CN115147375B (zh) * | 2022-07-04 | 2023-07-25 | 河海大学 | 基于多尺度注意力的混凝土表面缺陷特征检测方法 |
CN115937567B (zh) * | 2022-09-07 | 2023-07-07 | 北京交通大学 | 一种基于小波散射网络和ViT的图像分类方法 |
CN115588166B (zh) * | 2022-11-10 | 2023-02-17 | 新乡市诚德能源科技装备有限公司 | 一种防泄漏船用lng燃料罐 |
CN116051473A (zh) * | 2022-12-21 | 2023-05-02 | 中国石油大学(北京) | 焊缝缺陷识别模型建立方法及装置、识别方法及装置 |
CN116610080B (zh) * | 2023-05-23 | 2023-11-10 | 浙江众邦家居有限公司 | 休闲椅的智能生产方法及其控制系统 |
CN116503865A (zh) * | 2023-05-29 | 2023-07-28 | 北京石油化工学院 | 氢气道路运输风险识别方法、装置、电子设备及存储介质 |
CN116778346B (zh) * | 2023-08-23 | 2023-12-08 | 蓝茵建筑数据科技(上海)有限公司 | 一种基于改进自注意力机制的管线识别方法及系统 |
CN117408996B (zh) * | 2023-12-13 | 2024-04-19 | 山东锋士信息技术有限公司 | 基于缺陷专注和边缘权重损失的表面缺陷检测方法 |
CN117975176B (zh) * | 2024-04-02 | 2024-06-04 | 山东省计算中心(国家超级计算济南中心) | 一种基于工业动态视觉自我提示的钢铁缺陷检测方法 |
CN118365969B (zh) * | 2024-06-18 | 2024-09-03 | 广州市运通水务有限公司 | 基于智能机器人的全工况城市地下管涵检测方法及系统 |
Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111598860A (zh) * | 2020-05-13 | 2020-08-28 | 河北工业大学 | 基于嵌入自注意力门模块的yolov3网络的锂电池缺陷检测方法 |
CN113822885A (zh) * | 2021-11-23 | 2021-12-21 | 常州微亿智造科技有限公司 | 融合多注意力机制的工件缺陷检测方法和装置 |
Family Cites Families (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111259930B (zh) * | 2020-01-09 | 2023-04-25 | 南京信息工程大学 | 自适应注意力指导机制的一般性目标检测方法 |
CN113496217B (zh) * | 2021-07-08 | 2022-06-21 | 河北工业大学 | 视频图像序列中人脸微表情识别方法 |
CN113673594B (zh) * | 2021-08-18 | 2022-07-12 | 浙江工商大学 | 一种基于深度学习网络的瑕疵点识别方法 |
-
2022
- 2022-01-26 CN CN202210090694.7A patent/CN114445366B/zh active Active
Patent Citations (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111598860A (zh) * | 2020-05-13 | 2020-08-28 | 河北工业大学 | 基于嵌入自注意力门模块的yolov3网络的锂电池缺陷检测方法 |
CN113822885A (zh) * | 2021-11-23 | 2021-12-21 | 常州微亿智造科技有限公司 | 融合多注意力机制的工件缺陷检测方法和装置 |
Also Published As
Publication number | Publication date |
---|---|
CN114445366A (zh) | 2022-05-06 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN114445366B (zh) | 基于自注意力网络的长输管道射线影像缺陷智能识别方法 | |
Nguyen et al. | Inspecting method for defective casting products with convolutional neural network (CNN) | |
Mery | Aluminum casting inspection using deep learning: a method based on convolutional neural networks | |
CN110648305B (zh) | 工业图像检测方法、系统与计算机可读记录介质 | |
CN110490842B (zh) | 一种基于深度学习的带钢表面缺陷检测方法 | |
CN117173461B (zh) | 一种多视觉任务的灌装容器缺陷检测方法、系统及介质 | |
CN114943739A (zh) | 一种铝管质量检测方法 | |
CN113344903A (zh) | 一种基于变分自编码器的玻璃瑕疵检测方法 | |
Fioravanti et al. | A deep artificial immune system to detect weld defects in DWDI radiographic images of petroleum pipes | |
CN113643268A (zh) | 基于深度学习的工业制品缺陷质检方法、装置及存储介质 | |
US20240087105A1 (en) | Systems and Methods for Paint Defect Detection Using Machine Learning | |
CN113111875A (zh) | 一种基于深度学习的无缝钢轨焊缝缺陷识别装置及方法 | |
CN116309277A (zh) | 基于深度学习的钢材检测方法及系统 | |
Jalayer et al. | Automatic visual inspection of rare defects: A framework based on gp-wgan and enhanced faster R-CNN | |
CN115070780A (zh) | 基于数字孪生的工业机器人抓取方法、装置及存储介质 | |
CN115861226A (zh) | 基于特征值梯度变化的深度神经网络智能识别表面缺陷方法 | |
Jacobsen et al. | A comparison between neural networks and decision trees | |
Chetoui et al. | Object detection model-based quality inspection using a deep CNN | |
CN116228682A (zh) | 一种基于多支路卷积神经网络的射线图像焊缝缺陷识别方法 | |
Dong | A pixel-wise framework based on convolutional neural network for surface defect detection | |
Naddaf-Sh et al. | Next-generation of weld quality assessment using deep learning and digital radiography | |
Oz et al. | A nested autoencoder approach to automated defect inspection on textured surfaces | |
Mustafaev et al. | Enhancing Metal Surface Defect Recognition through Image Patching and Synthetic Defect Generation | |
Sedaghat et al. | Automatic defects detection using neighborhood windows features in tire X-ray images | |
Kim et al. | Automated end-of-line quality assurance with visual inspection and convolutional neural networks |
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 |