[发明专利]一种基于深度学习的带钢缺陷检测方法

说明书

本发明公开了一种基于深度学习的带钢缺陷检测方法，基于U‑Net的编码器，编码器通过反复的卷积操作提取图像中的特征，并进行五次下采样操作，提取不同尺度的特征。解码器部分与编码器对称，接受了编码器的输出，进行五次上采样，并且每次上采样得到的结果会与编码器同尺度的特征图进行拼接。下采样前使用高斯低通滤波器，去除其中的高频信息，防止因特征图原频率过高，出现混叠现象，导致特征图出现偏差。被去除的高频信息中蕴含了原图的细节信息，这些细节信息对带钢小型缺陷的检测至关重要，直接舍弃会影响效果。本方法将这些高频信息进一步加回到模型解码器的特征图上，产生更精细的分割效果。

技术领域

本发明属于计算机视觉技术领域以及缺陷检测领域。其中主要涉及的知识包括一些数字图像处理、图像增强、图像语义分割方法、深度学习图像分割方法等。

背景技术

自工业革命时代起，钢铁材料的产能开始飞速提升，并逐渐成为人类社会中最为广泛应用的材料之一。在现代，一个国家钢铁行业的强盛程度，是其能否在21世纪迅速发展的基础物质保证。钢铁材料在国家工业需求、住房需求、交通需求等诸多方面具有不可或缺的作用。带钢则又是钢铁企业的重要产出物品之一。带钢是一种又窄又长的钢板,在汽车、航空航天、机械制造、建筑、日用五金等方面用途广泛，是产量大、用处广、种类多的钢材。带钢质量的好坏对最终产品的性能与质量有着很大影响。随着时代发展，各个商家对带钢的质量要求也在不断提高，行业竞争日趋白热化。对于钢铁企业来说，为了保证钢铁的生产质量，质量检测水平的提升是必不可少的。

在传统的带钢缺陷检测方法里，人工目测虽是一种有效的方法，但该劳动强度大，带钢的生产环境恶略，容易对工作人员的身体和心理造成极大伤害。目前的自动化检测方法中，涡流检测是一种很常见的方法。该方法的缺陷检测方法主要是使用高频感应线圈让带钢内产生感应电流，只要检测表面温度变化，就可以检测出带钢的表面缺陷。但该方法对被测物表面状态的要求较高，对部分类型的缺陷检测效果并不好。

除此之外，随着图像采集技术的发展，基于计算机视觉的方法也被投入使用。该方法主要是将带钢样品的表面扫描为数字图像，进行进一步的检测。在计算机数字图像处理领域内，带钢的缺陷检测问题可以被视为一种图像的语义分割问题，即将图像中的每一个像素分类为一种类别，将缺陷的区域分割出来。传统的图像语义分割方法主要是根据图像的材质、纹理和颜色等特征，结合专业知识，人工设计图像特征的提取方法，并使用支持向量机(SVM)、随机森林、AdaBoost等分类算法进行分类。显然，对于复杂的图像场景，人工设计特征的提取方法效果并不理想。

随着计算机运算能力的不断提升以及机器学习相关算法的不断积累，深度学习的相关技术得到了极大的发展。深度学习在图像领域的研究很早之前就有过相关的尝试和应用。1998年，Lecun等人就提出了一种结合了卷积操作的神经网络，该网络被叫做卷积神经网络(CNN,Convolutional Neural Network)。起初，在简单的小尺寸图像场景下，CNN取得了很好的效果，但却很难处理大型图片。直到2012年，Krizhevsky等人构建了更深的网络结构AlexNet，该网络在著名的ImageNet数据集上取得了极大的进展，这使得深度学习在图像识别领域的研究工作大幅向前迈进。2015年He等人提出了深度残差网络(ResNet)，ResNet引进了跳跃式连接的结构，使得模块输入可以直接加到模块的输出上，避免了模型的退化现象，网络深度也因此达到了前所未有的152层，这种跳跃式连接的结构也基本成为今后CNN设计的基础单元。在语义分割领域，Long等人在2014年设计处理了全卷积神经网络(FCN)，该模型可以直接完成端到端的图像语义分割，极大的简化语义分割相关问题的复杂度，取得了突破性进展了突破性进展。在随后的几年中，SegNet，U-Net，DeepLab、PSPNet等其他语义分割模型相继提出，并在语义分割领域大放异彩。