基于深度学习的磁环表面缺陷检测算法

基于深度学习的磁环表面缺陷检测算法

人工智能技术与咨询 

来源：《 人工智能与机器人研究》 ，作者罗菁等

关键词: 缺陷检测；深度学习；磁环；YOLOv3；

摘要: 在磁环的生产制造过程中，常常由于生产环境、制造工艺等因素，难免会使磁环表面出现各种类型的缺陷。针对传统人工检测低效、耗时、检测精度低的缺点，本文提出了一种基于YOLOv3的磁环表面缺陷检测方法。实验结果表明，YOLOv3的平均识别精度达到了96.19%，单张图片检测速度达到了24.46 ms，该方法在磁环缺陷检测上有一定的先进性和有效性。

1. 引言

磁环是呈圆柱形的径向磁体，生活中广泛应用在电子定时器、汽车制造以及仪表器件等行业的各类电器装置，它的质量直接影响所在装置的性能。但是在目前的生产过程中，由于原料组成、加工工艺和设备条件的影响，磁环表面不可避免地会出现一些加工缺陷，如缺口、污点等，这些缺陷会影响到磁环的使用寿命和性能。目前，磁环表面一些常见的缺陷主要由有一定经验的工人检测，用肉眼观察磁环的表面是否有缺陷。这种检测方法效率低，容易出现视觉疲劳，造成误差 [1]。

为了克服人工缺陷检测的不足，很多人提出了各种表面缺陷检测方法，可以分为两类：传统的机器视觉检测方法和深度学习检测方法。李雪琴等 [2] 提出了一种非下采样轮廓域自适应阈值表面的磁瓦缺陷自动检测方法。林丽君等 [3] 提出了一种结合图像加权信息熵和小波模极大值的磁瓦表面裂纹检测算法。这些方法对于特定缺陷具有速度快、精度高的优点，但受光照和人工的影响较大。

近年来，随着深度学习的广泛应用，已在目标检测 [4] 领域取得了一系列成果，在工业检测领域得到了很大的应用。目前，广泛应用的深度学习目标检测算法可分为单阶段检测算法和两阶段检测算法。单阶算法直接生成对象的类别概率和位置坐标值，其代表性算法为SSD [5] 和YOLO [6] [7] [8] 网络。两阶段算法将检测分为两个阶段，首先生成候选区域，然后对候选区域进行分类。代表性算法有R-CNN [9] 及其改进算法Fast R-CNN [10] 和Faster R-CNN [11] 等。

一般情况下，两阶段算法在准确度上有优势，而单阶段算法在检测的速度上有优势。由于单阶段算法在速度上更有优势，所以更受到工程化的青睐。本文也将使用YOLOv3网络对磁环表面缺陷进行检测，为了提升检测速度和精度使用k-means聚类方法重新选取先验框，保证检测的实时性。

2. 检测算法

2.1. YOLOv3网络

YOLOv3是在YOLO和YOLOv2算法的改进基础上，主要用于图像目标检测、视