基于支持向量机的金属表面缺陷检测

2021-05-11 19:57缪徐超张旭徐洋余前溪程凯强
智能计算机与应用 2021年2期
关键词:支持向量机特征提取

缪徐超 张旭 徐洋 余前溪 程凯强

摘要:针对人工检测金属表面缺陷效率低、主观意识强、无法长时间工作等缺点,提出一种基于支持向量机监督检测、分类以及测量的金属表面缺陷的方法,并使用Matlab软件设计一个图形用户界面(GUI),便于检测人员使用。研究中先对工厂采集的图像进行Gabor滤波和对比度增强的前处理。然后使用方向梯度直方图(HOG)和灰度共生矩阵(GLCM)进行特征提取,为后续提高检测准确率打下基础。最后对于每种缺陷类型都采用270张图片进行模型训练,利用训练好的模型对测试图片进行测试。实验结果表明,最终3种缺陷类型的总正确率为88.9%,该检测方法能够有效地检测出金属表面缺陷。

关键词:表面缺陷检测;支持向量机;图片前处理;特征提取

【Abstract】Aimingattheshortcomingsofmanualdetectionofmetalsurfacedefects,suchaslowefficiency,strongsubjectiveawareness,andinabilitytoworkforalongtime,amethodbasedonSupportVectorMachinesuperviseddetection,classificationandmeasurementofmetalsurfacedefectsisproposed,andMatlabsoftwareisusedforGUIpackagingforeasydetectionofthepersonnel.TheresearchfirstperformsGaborfilteringandcontrastenhancementpre-processingontheimagescollectedbythefactory,thenusesdirectionalgradienthistogram(HOG)andgraylevelco-occurrencematrix(GLCM)forfeatureextractiontolaythefoundationforsubsequentimprovementofdetectionaccuracy.Finally,foreachdefecttype,270picturesareusedformodeltraining,andthetrainedmodelisusedtotestthetestpictures.Experimentalresultsshowthatthefinaltotalaccuracyofthethreetypesofdefectsis88.9%,andthisdetectionmethodcaneffectivelydetectmetalsurfacedefects.

【Keywords】surfacedefectdetection;SupportVectorMachines;picturepre-processing;featureextraction

作者簡介:缪徐超(1996-),男,硕士研究生,主要研究方向:切削原理与应用;张旭(1978-),女,博士,教授,主要研究方向:图像处理和切削原理与应用;徐洋(1995-),男,硕士研究生,主要研究方向:图像处理;余前溪(1994-),男,硕士研究生,主要研究方向:切削原理与应用;程凯强(1996-),男,硕士研究生,主要研究方向:图像处理。

0引言

金属表面缺陷检测在许多工业领域(如飞机机身、核电站蒸汽发生管道和钢桥等)中非常重要。目前,有关金属表面缺陷的机器学习检测技术发展迅速。机器学习检测是采用图像处理的方法对金属表面进行分析处理,检测金属表面是否有缺陷。对表面有缺陷的产品和正在使用中的零件,予以去除和更换,避免在后续的使用过程中造成安全事故。在此背景下,过去几十年里大量学者致力于研究金属表面缺陷检测技术。

McCluskey等人[1]对金属薄膜电容器表面缺陷的检测方法进行了研究并运用于实际。研究中将检测方法分为3步:进行产品可靠性和成熟度分析;如果有可能,通过非破坏性分析检测缺陷;开发电气测试以筛除早期故障。Qi等人[2]为了对基于工业CT的金属材料增材制造中内部微小缺陷进行精确检测,提出了一种基于CT图像中微小缺陷灰度变化系数的金属材料微小缺陷检测的方法。实验结果表明,该缺陷检测方法相对误差控制在4%以内。Feng等人[3]提出了一种基于高动态范围成像技术的缺陷检测方法,解决了工业零件高反射面外观检测问题。智能工厂的创建开启了一项研究,该研究通过分析从物联网收集的数据来实时预测产品质量。这项研究吸引了Yun等人[4]的注意,因此提出了一个基于视觉缺陷检测系统来检测金属表面缺陷。该项研究提出了一种新的卷积变分自编码器和卷积神经网络的缺陷分类算法用于金属表面缺陷检测,使用从实际金属生产线获得的缺陷图像进行实验,证明了该方法的良好性能。对于人工检测金属表面缺陷工作量大,效率低的情况。刘建春等人[5]在使用机器视觉对金属表面缺陷检测前,根据金属零件表面反光的特点对图像进行有效的前处理,实验结果表明,经过前处理的图片能够有效检测出金属边缘缺陷。金闳奇等人[6]对于检测金属表面缺陷效率不高的问题,提出了一种多尺度的检测方法。实验结果显示,该方法检测单张图片用时约为13ms。王俊强等人[7]以OpenCV软件为基础,开发了带钢表面缺陷检测系统,克服了采用人工检测方式的效率低、带有主观意见、无法持续长时间工作等缺点。Zhang等人[8]用微波无损检测技术对厚涂层下缺陷的形状和尺寸进行了实验研究。使用主成分分析算法来提高方法的分辨率,对已知缺陷的钢板和未知缺陷的涂层钢板在不同涂层厚度下进行了测试。结果表明,该方法可以提高涂层钢板缺陷检测能力。Layouni等人[9]为解决人工分析从金属缺陷的磁扫描中收集到的图案的大量数据比较困难的问题,提出了一种自动分析的方法,通过使用自适应小波算法来检测和估计金属缺陷的长度。该方法效率高、精度高,适用于多种类型的缺陷形状。

基于此,本文提出了基于支持向量机的缺陷检测、分类和测量,该技术识别缺陷效率高,错误率小。使用MatlabGUI进行了封装,方便金属缺陷检测人员使用。

1图像前处理和特征提取

1.1图像前处理

在对金属表面缺陷图像进行检测、分类和测量之前要先进行前处理,因为工业环境比较复杂,光线、室温、空气湿度等都会对摄像头成像的图片造成一定的噪声干扰。本次研究采用了2个方法对图片进行前处理:

(1)对比度增强。在很多有关图像处理的算法中,对比度增强是后续处理的基础,有利于更好地区分前景和背景对象,提高识别的正确率,因此是很重要的一步,本文将摄像头采集的图像进行灰度增强,具体方法是:

1.2特征提取

在利用SVM进行金属表面缺陷图片的分类检测前,特征提取是获得较好检测结果的基础,对于图像的检测、分类和测量起着至关重要的作用。这里对前处理好的图像进行了2次特征提取,拟采用方法和设计流程如下。

(1)方向梯度直方图(HOG)特征是一种在计算机视觉和图像处理中用来进行物体检测的特征描述子,通过计算和统计图像局部区域的梯度方向直方图来构成特征,可以将局部目标的表象和形状通过梯度方向分布描述出来。

①每个像素点的梯度通过2个算子,即:(-1,0,1)和(-1,0,1)转置进行计算。

②Bin数量采用9,每个通道为40°。可以得到Cell单元的方向直方图,其中Cell尺寸是4*4。

③本文Block尺寸为2*2,将Cell单元合并组合为更大的block区间,并在block做归一化,最终特征的长度是86436。

(2)GLCM特征,即灰度共生矩阵特征。能反映图像灰度关于方向、相邻间隔、变化幅度等综合信息,是分析图像的局部模式和排列规则的基础。由于GLCM可以产生14种特征,而考虑到减少计算量,减少冗余特征,本文采用了对比度、相关、能量、均匀性四个特征。通过在0°、45°、90°、135°四个方向进行计算,求出4个方向矩阵的特征值后,将4个特征值的平均值作为最终特征值共生矩阵。最终特征的长度是80。

2支持向量机

在对图像进行前处理和特征提取后得到有效的图像特征,若再能设计一个好的检测器就显得尤为重要。支持向量机是一种描述响应与一个或多个描述符之间非线性关系的方法。在解决小样本、非线性和高位模式识别问题上体现出优势。与响应具有正态分布的简单线性回归或指数回归不同,支持向量机的响应可以是正态、二项或泊松。支持向量机采用函数f进行线性描述。在后续的预测分析中,响应具有正态分布。支持向量机正则化是一种通过附加惩罚项来限制模型复杂度的收缩方法,可以识别重要的描述符,选择描述符,并在模型公式中產生更少的系数。在支持向量机拟合回归中,首先引入非线性映射x,将其映射到m维特征空间。然后在特征空间中构建线性模型,得到:

3图像检测结果及分析

利用SVM进行图像检测主要分为2步:

(1)SVM训练,导入训练图片,进行图像前处理和特征提取,并输入模型进行训练。

(2)SVM检测,本文采用SVM进行金属表面缺陷的检测。同样地,对测试图片进行前处理和特征提取,输入训练好的模型及自动输出预测结果。GUI界面如图1所示。

在本次模型训练过程中,对于每组缺陷,研究中都采用270张样本进行训练。在检测过程中对30张图片进行缺陷的检测,最终3种损伤类型的总识别正确率为88.9%。

金属表面氧化皮缺陷图片检测结果如图2所示。GUI界面检测结果中显示出缺陷种类为氧化皮,并显示有4处缺陷。

划痕检测结果如图3所示,结果中显示出缺陷种类为划痕,并显示共有2条划痕。

Fig.3Defectdetectionresultsofscratch

斑点检测结果如图4所示,结果中表明缺陷的种类为斑点,并显示出斑点缺陷所占图片中金属面积的2.9442%。

Fig.4Defectdetectionresultsofspots

4结束语

本文采用支持向量机的方法对金属表面缺陷图片进行了分类和测量。在检测之前对采集的图片进行了对比度增强和Gabor滤波预处理,降低了拍摄图片时复杂工业环境产生的噪声。并进行了特征提取,保证了较高的SVM分类识别率。实验结果表明,使用本文提出GUI界面可以方便快速地取得较好的金属表面缺陷分类测量效果。

参考文献

[1]MCCLUSKEYFP,LINM,MENGOTTIE.Eliminatinginfantmortalityinmetallizedfilmcapacitorsbydefectdetection[J].MicroelectronicsReliability,2014,54(9-10):1818-1822.

[2]QIZicheng,NIPeijun,JIANGWei.QuantitativedetectionofminordefectsinmetalmaterialsbasedonvariationcoefficientofCTimage[J].Optik,2020,223:165269.

[3]FENGWei,LIUHongdi,ZHAODaxing,etal.Researchondefectdetectionmethodforhigh-reflective-metalsurfacebasedonhighdynamicrangeimaging[J].Optik,2020,206:164349.

[4]YUNJP,SHINWC,KOOG,etal.Automateddefectinspectionsystemformetalsurfacesbasedondeeplearninganddataaugmentation[J].JournalofManufacturingSystems,2020,55:317-324.

[5]刘建春,林海森,黄勇杰,等.基于机器视觉的金属边缘细微缺陷检测方法的研究[J].制造技术与机床,2018(11):137-140.

[6]金闳奇,陈新度,吴磊.基于特征金字塔的多尺度金属表面缺陷检测[J].组合机床与自动化加工技术,2020(8):97-100.

[7]王俊强,张业宏,马冲,等.基于机器视觉的带钢表面缺陷检测系统[J].邢台职业技术学院学报,2020,37(3):78-82.

[8]ZHANGHong,GAOBin,TIANGuiyun,etal.MetaldefectssizinganddetectionunderthickcoatingusingmicrowaveNDT[J].NDT&EInternational,2013,60:52-61.

[9]LAYOUNIM,HAMDIMS,TAHARS.Detectionandsizingofmetal-lossdefectsinoilandgaspipelinesusingpattern-adaptedwaveletsandmachinelearning[J].AppliedSoftComputing,2017,52:247-261.

[10]张立新.基于表面缺陷检测的金属图像识别[J].粘接,2019,40(8):121-123.

猜你喜欢
支持向量机特征提取
基于MED—MOMEDA的风电齿轮箱复合故障特征提取研究
基于曲率局部二值模式的深度图像手势特征提取
一种针对特定无线电信号的识别方法
基于改进支持向量机的船舶纵摇预报模型
基于模糊K近邻的语音情感识别
基于SVM的烟草销售量预测
动态场景中的视觉目标识别方法分析
论提高装备故障预测准确度的方法途径
基于熵技术的公共事业费最优组合预测
基于支持向量机的金融数据分析研究