高级检索+

基于迁移学习的磁瓦缺陷分类方法

Magnetic Tile Defect Classification Method Based on Transfer Learning

  • 摘要: 针对磁瓦缺陷检测时存在的缺陷样本量少、类别不平衡、模型训练过拟合等问题,提出了一种基于迁移学习的磁瓦缺陷分类方法.该方法在大型图像数据集ImageNet上预训练深层卷积神经网络VGG16,然后使用迁移学习方法将模型迁移到磁瓦缺陷分类研究中,先冻结模型前几层的参数,再用磁瓦缺陷数据集训练调整后的全连接层,并在测试集上测试模型的分类效果.实验结果显示,6类磁瓦缺陷识别准确率达到了98.69%,明显高于人工分类精度和传统机器视觉分类方法识别精度.该方法实现了较高的磁瓦缺陷分类准确率,同时也极大缩短了训练时间,为工业生产中的实际应用提供了可靠的依据.

     

    Abstract: A fast classification method based on transfer learning is proposed for detecting defects in magnetic tile. This method pre-trains the deep convolutional neural network VGG16 on the large image dataset ImageNet and then uses the transfer learning method to transfer the model to the study of magnetic tile defect classification. First, freeze the parameters of the first few layers of the model.Secondly, the magnetic tile defect dataset is used for training and adjustment of fully connected layer.Finally, test the classification effect of the model on the test set. Experimental results show that the recognition accuracy of the six types of magnetic tile defects reached 98.69%, which is significantly higher than the recognition accuracy of manual classification and traditional machine vision classification methods. The superiority of this method provides a reliable basis for practical applications in industrial production.

     

/

返回文章
返回