基于轻量级卷积神经网络的实时缺陷检测方法研究

Research On Real-time Defect Detection Method Based On Lightweight Convolutional Neural Network

摘要:

应用机器视觉实现磁片表面缺陷的自动检测可以提高生产效率、降低生产成本。深度卷积神经网络具有高精度的分类性能,尤其在图像识别方面有显著的优点。但是目前提出的深度神经网络模型,由于参数量和计算量的巨大,在工业生产流水线上不能满足实时检测的需求。针对这个问题,基于深度可分离卷积和通道混洗,提出了一种轻量级高效低延时的卷积神经网络架构MagnetNets。为了评估MagnetNets网络模型的性能,将MagnetNets网络模型与MobileNets、ShuffleNet、Xception、MobileNetV2在公开数据集ImageNet中做了对比实验。然后将MagnetNets网络模型应用在磁片缺陷检测系统中进行缺陷检测。实验结果表明,提出的网络架构显著地减少参数数量,具有良好的性能。同时在磁片缺陷检测系统中减少了延时,提高检测速度,缺陷检测识别率达到了97.3%。

Abstract:

The application of machine vision to the automatic detection of surface defects on magnetic sheets can increase production efficiency and reduce production costs. Deep convolutional neural networks have high-precision classification performance, especially in image recognition. However, the deep neural network model proposed so far cannot meet the requirements of real-time detection in the industrial production line due to the huge amount of parameters and computation. To solve this problem, based on deep separable convolution and channel shuffling, we proposed a lightweight, high-efficiency and low-latency convolutional neural network architecture called MagnetNets.In order to evaluate the performance of the MagnetNets network model, we compared it with MobileNets, ShuffleNet, Xception, and MobileNetV2 in the public dataset ImageNet.And then the MagnetNets network model is applied to the defect detection system for magnetic defect detection.The experimental results show that the proposed network architecture significantly reduces the number of parameters and has good performance,At the same time, the delay is reduced and the detection speed is improved in the disk defect detection system and the defect detection recognition rate reaches 97.3%.