基于卷积神经网络的织物表面缺陷分类方法 |
| |
引用本文: | 景军锋,刘娆.基于卷积神经网络的织物表面缺陷分类方法[J].测控技术,2018,37(9):20-25. |
| |
作者姓名: | 景军锋 刘娆 |
| |
作者单位: | 西安工程大学 电子信息学院,西安工程大学 电子信息学院 |
| |
基金项目: | 国家自然科学基金项目资助(61301276) |
| |
摘 要: | 针对织物缺陷检测时疵点种类繁多且传统人工检测方法漏检率高的问题,提出了一种基于卷积神经网络的织物表面缺陷分类方法。因卷积神经网络(CNN)训练时参数多、样本量大,且极易陷入过拟合,利用微调卷积神经网络模型Alexnet对织物疵点图像进行特征提取,初始化采用原网络的参数而非随机初始化参数;再针对特定目标下的训练样本对网络参数进行微调;最后利用softmax回归算法进行预测分类。分别用三种方法和两种织物进行测试,结果表明:针对特定目标微调后的Alexnet网络,在两类织物测试中均能达到95%以上的分类准确率。
|
关 键 词: | 卷积神经网络 织物缺陷分类 Alexnet 迁移学习 |
Classification Method of Fabric Surface Defects Based on Convolution Neural Network |
| |
Abstract: | A new method for classification of fabric surface defects based on convolution neural networks is proposed,which is mainly used to solve a wide range of defects and the high error rate of manual detection method.Convolution neural networks need to train a large number of parameters,the training process requires a large number of samples,and it is easy to fall into over-fitting,so the fine-tuning Alexnet network model is adopted for feature extraction.The original network parameters were used to initialize the network,and then fine-tune the parameters for a specific sample.Finally,the softmax algorithm was used for prediction and classification.The experiment was carried out using three methods and tested with two types of fabrics.The results show that the fine-tuned Alexnet can achieve more than 95% classification accuracy in both types of fabric tests. |
| |
Keywords: | convolution neural network(CNN) fabric defect classification Alexnet transfer learning |
|
| 点击此处可从《测控技术》浏览原始摘要信息 |
|
点击此处可从《测控技术》下载全文 |