[发明专利]一种基于多尺度转移的充电器表面缺陷检测方法

说明书

本发明涉及一种基于多尺度转移的充电器表面缺陷检测方法，包括步骤：数据预处理；搭建稠密卷积神经网络；搭建多尺度转移模型；得到的不同尺度的特征图输入多尺度缺陷目标检测模型中，多尺度缺陷目标检测模型中的多尺度缺陷目标检测网络由分类子网络和边界回归子网络组成。本发明的有益效果是：本发明使用基于稠密卷积神经网络的多尺度转移算法，搭建了一种端到端的多尺度目标检测模型，应用多特征匹配的方法，可以对电子设备的充电器外表面进行自动且高效的缺陷目标检测；图像级标签缺省的手机屏幕缺陷检测对于工业应用具有重大应用价值，可以节约企业生产成本，并保护消费者权益。

技术领域

本发明属于缺陷检测技术领域，具体涉及一种基于多尺度转移的充电器表面缺陷检测方法，用以解决充电器表面缺陷检测问题。

背景技术

随着计算机技术、通信技术以及互联网技术的高速发展，电子设备在人们生产生活中所扮演的角色越来越重要，而配套的充电器更是不可或缺的。为了保护消费者权益，以及提高品牌价值，防止存在缺陷的充电器流入市场，在充电器出厂前，对充电器的质量进行检测是必需的过程，而对其外观表面的缺陷检测则是一个研究重点。

传统的检测手段是在生产线上安排工人值守，工人对生产线上的屏幕用肉眼逐一进行检测。但是这种手段存在检测效率慢、人工成本高、缺乏统一的判断标准等缺点。

近年来，随着机器学习、深度学习在计算机视觉、模式识别等领域大放异彩，对于充电器外观表面缺陷检测又提供了另外的强有力的方法。目标检测要求在给定的图像中定位所要提取的对象类别的所有实例，基于卷积神经网络(CNN)的检测框架在速度和精度方面取得了显著进展。在其中，尺度问题是缺陷目标检测的核心问题，为了检测不同尺度的目标，一种基本策略是利用图像特征图谱获取不同尺度的特征。而由于CNN中每一层的感受野是固定的，所以充电器缺陷图像中固定的感受野与不同尺度的缺陷存在不一致。并且，浅层特征图有小的感受野，用于检测小的目标，而深层特征地图有大的感受野，用于检测大的缺陷目标。然而，浅层特征的语义信息较少，这可能会影响小目标的缺陷检测的性能。

发明内容

本发明的目的是克服现有技术中的不足，提供一种基于多尺度转移的充电器表面缺陷检测方法。

这种基于多尺度转移的充电器表面缺陷检测方法，包括以下步骤：

步骤1、数据预处理：在输入模型前，将充电器外表面缺陷图像分别进行尺度放缩、直方图均衡、均值平滑和图像校正；

步骤2、搭建稠密卷积神经网络：将步骤1预处理得到的充电器外表面缺陷图像输入稠密卷积神经网络中进行特征提取，并在每一个稠密单元层中获取不同的特征；

步骤3、搭建多尺度转移模型；将步骤2中每一个卷积单元层不同尺度的输出特征输入多尺度转移模型中，获取不同尺度的特征图，结合不同分辨率的特征图进行多尺度的缺陷目标检测；

步骤3.1、利用稠密卷积神经网络的网络结构得到强语义特征映射，通过concat操作将底层特征直接转移到稠密卷积神经网络的顶层；

步骤3.2、在多尺度转移模型中使用平均池化层来获得低分辨率的特征图；使用尺度转移层技术获得高分辨率的特征图；将多尺度转移模型直接嵌入稠密卷积神经网络中，从稠密卷积神经网络的最后一个稠密卷积单元层得到不同尺度的特征图；

步骤3.3、利用步骤3.2所得不同尺度的特征图的特征来映射构造多尺度缺陷目标检测模型；

步骤4、将步骤3中得到的不同尺度的特征图输入多尺度缺陷目标检测模型中，多尺度缺陷目标检测模型中的多尺度缺陷目标检测网络由分类子网络和边界回归子网络组成；分类子网络对多尺度缺陷目标进行目标分类，边界回归子网络对多尺度缺陷目标进行边界位置回归分析。

作为优选，步骤1具体包括如下步骤：

步骤1.1、将充电器外表面缺陷图像进行尺度放缩，达到设定的尺度大小；