[发明专利]基于BP神经网络和支持向量机的铝板表面缺陷分类方法

说明书

技术领域

本发明涉及模式识别技术领域，尤其涉及一种基于BP神经网络和支持向量机的铝板表面缺陷分类方法。适用于金属表面缺陷检测系统对含有油斑等多类金属表面缺陷进行识别分类。

背景技术

冷轧铝板表面缺陷检测分类经历了人工目测分类、传统无损检测分类和基于机器视觉的检测分类3个发展阶段，人工目视检测分类表面缺陷的方法效率低、容易漏检、劳动强度大、实时性差。传统无损检测分类方法包括涡流检测分类、红外检测分类、漏磁检测分类、激光检测分类等,这些方法检出的缺陷种类少,并且检测分类的实时性不强,无法有效评估产品的表面质量状况。目前,基于机器视觉的表面质量检测分类方法是研究的热点,该方法采用CCD摄相机采集铝板表面图像,然后通过图像处理和分析提取缺陷图像特征。进行缺陷的自动分类。冷轧铝板表面缺陷的分类实质是模式识别，传统的模式识别或人工神经网络方法适合少类别分类，对多类别分类总体识别精度不高，算法耗时多，泛化能力差且需要较多的训练样本。支持向量机适合多类别分类，整体分辨率较高，但是对于个别类缺陷识别率不是很高，尤其是油斑。因此提高油斑缺陷识别分类率和整体识别分类率成为亟待解决的问题。

提高油斑缺陷识别分类率和整体识别分类率可以在硬件上进行改进，使铝板表面缺陷检测系统采用超高性能的CCD相机，并且通过改变硬件设施解决铝板表面高反光性对缺陷正确检测的影响。这种方案在一定程度上可以提高冷轧铝板表面单个缺陷识别分类率和整体识别分类率，却增加了硬件的复杂性，提高了成本，不利于在需求铝板表面缺陷检测系统的中小型冷轧铝板生产企业进行推广使用。

发明内容

本发明的目的就是为了解决上述问题，提供了一种基于BP神经网络和支持向量机的铝板表面缺陷分类方法，该方法不仅提高了冷轧铝板表面油斑的识别分类率，同时也提高了冷轧铝板表面缺陷的整体识别率，也可用于其他金属表面缺陷的识别分类，且简单，易于实现。

为了实现上述目的，本发明采用如下技术方案：

基于BP神经网络和支持向量机的铝板表面缺陷分类方法，利用神经网络的非线性映射功能，选择多输入双输出的前向神经网络，从冷轧铝板表面缺陷上提取的缺陷特征值作为学习样本数据，应用误差反向传播学习算法对网络进行训练，得到最优的BP神经网络分类模型，实现对油斑的高识别分类。

把BP神经网络分类模型认为不是油斑的第一类缺陷交由支持向量机进行再分类，支持向量机采用一对一的分类方法构建多个二分类支持向量机模型进行分类，每个支持向量机用所分的二分类所对应提取的缺陷特征值进行训练，训练完成后支持向量机分类模型则对BP神经网络认为不是油斑类的第一类缺陷进行分类；综合BP神经网络和支持向量机分类的结果输出，得到油斑及第一类缺陷的分类。

基于BP神经网络和支持向量机的铝板表面缺陷分类方法，提取铝板表面缺陷的特征值作为BP神经网络分类模型的输入量，油斑和非油斑的第一类缺陷作为输出量构建BP神经网络分类模型；

第一类缺陷采用一对一的分类方法，构建多个支持向量机分类模型；

获取学习样本并训练BP神经网络分类模型与支持向量机分类模型；

利用训练后的BP神经网络分类模型分出油斑和非油斑的第一类缺陷，把训练后的BP神经网络分类模型认为是油斑的测试样本剔除，只剩下第一类缺陷的测试样本；

用训练后的支持向量机分类模型对剩下的第一类缺陷再进行分类；最后统计得到分类结果；

利用BP神经网络分出油斑，再利用支持向量机对第一类缺陷进行细分，两类分类方法的融合实现了精准快速的分类。

所述第一类缺陷包括气泡、破皮、暗痕、亮痕、孔洞及黑线。

采用自适应阈值法对缺陷图进行分割，从几何特征、形状特征及灰度特征三类特征中提取得铝板表面缺陷的多个特征值，将每一个特征值归一化在[-1,1]之间，把归一化的特征值作为学习样本的数据。

获取学习样本的方法为：对铝板表面缺陷检测系统采集的缺陷图进行缺陷特征值数据的提取，得到学习样本数据。

得到学习样本后，采用BP算法训练BP神经网络分类模型并得到最优的BP神经网络分类模型参数。

最后的统计得到的分类类别包括油斑、气泡、破皮、暗痕、亮痕、孔洞及黑线。

BP神经网络分类模型的结构为三层，包括输入层、隐含层和输出层，输入层节点数与提取的缺陷特征值维数相同，输出层节点数为2，隐含层节点数为6～15。