基于卷积神经网络的竹片图像分类方法与流程

[0001]
本发明涉及一种竹片图像分类方法。属于图像识别技术领域。


背景技术：

[0002]
随着时代的发展，人们对自身的保养意识逐渐加强。炎炎夏日里，良好舒适的睡眠是人们健康的保证，竹席具有良好的的透气性、凉爽和不卷曲等特性，逐步取代草席成为家家户户度过夏天的的必备品。与此同时，炭化竹席具有能吸附灰尘、清新空气、除菌、防臭等功能，能够促进人体新陈代谢。麻将凉席由大量形似麻将的竹片编织而成，可以达到按摩全身的效果。可以看出竹席有许多优点，所以竹席的需求量长盛不衰。
[0003]
目前，中国的竹产业已形成了一种新的有潜力的行业，从资源培育与加工利用的潜力和活力，出口贸易，再到竹生态旅游。2009年，竹材产量13亿5600万，竹笋产量46.53多吨，竹产业总产值710亿元，中国竹制品出口额达15亿美元，产品出口到世界177个国家，居世界第一位。当前工业生产的所有麻将凉席竹片大多采用传统的人工筛选，筛选人员釆用肉眼观察。随着环境、工作时间和疲劳度所限，从众多竹片中发现缺陷，并完成分拣工作，这将耗费大量的人力、物力和财力，并且准确度和效率都较低。尤其是竹林是重要的非木质森林资源，我国现有竹子种类500余种。竹片的常见缺陷是各竹片间的形状差异，人工检测作为传统的检测方法，虽然能直观的观测到缺陷但是仍有一些缺点。首先，当缺陷图像与背景灰度差异较小或者缺陷不是很明显或者缺陷所在图像的背景比较模糊，人工检测会产生较大的误差；其次，人工检测很难跟上工业生产中竹片的传输速率，所以对产品检测的实时性会产生影响；最后，工人在进行人工检测的过程中或多或少的会受到主观因素影响导致缺陷判断的标准不能百分百相同，导致无法保证检测结果统一准确，并且经过长时间的人工检测，会对人的眼睛、身体等造成疲劳，导致漏检和错检。所以目前我们急需一种实时性好、可靠性高、智能的竹片表面缺陷检测识别分类技术。
[0004]
国内目前对于竹片检测的自动化设备的生产已经逐步扩大。针对竹片缺陷检测的研究，广西师范大学蒋贤明等人推出一款竹片分拣机，该机器侧重点是挑选出未炭化的竹片，对于竹片的其它缺陷检测不够准确，算法相对来说比较复杂且检测速率低。浙江理工大学的华于生等人提出了基于嵌入式的竹片表面缺陷监测系统，但是检测速度比较慢而且硬件配置比较落后；湖北工业大学陈张言等人提出基于机器视觉的竹片缺陷检测系统，检测准确率已经可以到90％以上但是对于炭化竹片来讲无法做到准确识别；广西师范大学王东旭等人提出基于bp神经网络的竹片正反面识别，准确率达到97％，但是耗时很长并且无法识别出竹片本身所存在的缺陷。


技术实现要素：

[0005]
本发明是为了解决现有的竹片检测方法存在竹片本身缺陷的检测准确率较低且耗时较长的问题。
[0006]
1、基于卷积神经网络的竹片图像分类方法，包括以下步骤：
[0007]
s1、采集竹片图像建立数据集；
[0008]
s2、将数据集分为训练集、验证集、测试集；
[0009]
针对多个卷积神经网络中的每个卷积神经网络模型，分别利用迁移学习方法进行实验，实验过程中，分别利用训练集、验证集确定每个卷积神经网络模型权重及调整模型的超参数，然后利用测试集检验每个卷积神经网络的泛化能力；
[0010]
通过混淆矩阵对测试集下每个卷积神经网络的准确率进行对比；确定识别准确率最高的一个卷积神经网络，作为竹片图像分类卷积神经网络；
[0011]
s3、采集待分类的竹片图像，利用竹片图像分类卷积神经网络进行竹片图像分类。
[0012]
进一步地，对测试集下每个卷积神经网络的准确率进行对比的过程是通过绘制的每个卷积神经网络实验结果折线图实现准确率进行对比的。
[0013]
进一步地，绘制的每个卷积神经网络实验结果折线图的过程是利用matlab软件完成的。
[0014]
进一步地，利用训练集、验证集确定每个卷积神经网络模型权重及调整模型的超参数之前的卷积神经网络模型的超参数如下：
[0015]
学习衰减率：0.001；动量：0.5；学习率：0.01；权值衰减率：0.0005；批量：32；轮数：60；训练算法：随机梯度下降法。
[0016]
进一步地，通过混淆矩阵对测试集下每个卷积神经网络的准确率进行对比的过程包括以下步骤：
[0017]
针对于每个卷积神经网络的预测结果，进行如下标记：
[0018]
预测值为1，记为p；预测值为0，记为n；
[0019]
预测值与真实值相同，记为t；预测值与真实值相反，记为f；
[0020]
tp表示真阳，即真实为1，预测为1；fn表示假阴，即真实为1，预测为0；fp表示假阳，即真实为0，预测为1；tn表示真阴，即真实为0，预测为0；
[0021]
利用准确率、精确率、召回率和f1-score评判混淆矩阵，其中准确率＝(tp+tn)/(tp+fp+tn+fn)，精确率＝tp/(tp+fp)，召回率＝tp/(tp+fn)，
[0022]
有益效果：
[0023]
本发明主要是利用普通深度学习和迁移学习技术中的3类不同结构的卷积神经网络进行实验，对自行采集的4类竹片图像进行自动图像分类识别，通过对比选择出识别效果最好的网络，最终效果最好的神经网络分类识别准确率达到98％以上，能很好的完成竹片筛检分类的任务，并且耗时短，6000多张图片只需要40分钟左右就可全部识别完毕，大大减少了人工成本的消耗。并且该发明是利用计算机对程序进行操控，大大减少了人工筛检的时间、精力、金钱，节省了成本，同时一次性分类图像可以达到成千上万张，节省时间。
附图说明
[0024]
图1为神经网络实验结果的折线图；
[0025]
图2为densenet121的混淆矩阵图。
具体实施方式
[0026]
具体实施方式一：
[0027]
本实施方式所述的基于卷积神经网络的竹片图像分类方法，包括以下步骤：
[0028]
s1、首先在保持同样的拍摄距离和角度的情况下对竹片进行图像采集，采集工具为魅族魅蓝e3系列手机，这样保证图像不因拍摄工具不同而有质量区别。数据集中给每张照片进行了编号，这样可以从编号确定照片中的竹片属于哪一种类，由于相机拍摄时不稳定可能导致图像模糊，在本实施方式中删除部分模糊化的图像。利用竹片图像建立数据集；
[0029]
对数据集内的每一张照片，进行手动裁剪，将每张照片尺寸改为统一的256
×
256，使图像更易于被处理。为了得到更加准确的识别分类效果，对采集到的4类竹片图像进行预处理，即利用直方图均衡增强方法对图像进行增强，直方图均衡增强是一种增强图像对比度的方法，其主要思想是将一副图像的直方图分布变成近似均匀分布，从而增强图像的对比度。
[0030]
s2、确定用于竹片图像分类的神经网络模型；
[0031]
为了找到效果最好的卷积神经网络，本实施方式选用了3种神经网络模型结构，分别是vgg16、resnet50以及densenet121。
[0032]
首先把输入的图像传输到卷积层中进行卷积，通过卷积操作对输入图像进行降维和特征抽取；但是经过卷积层提取特征以后的图像维数还是很高，此时需要池化层来进行处理，池化层的具体实现是在进行卷积操作之后对得到的特征图像进行分块，图像被划分成的不相交块，计算这些块内的最大值或平均值，得到池化后的图像，经过若干个卷积池化层对图像进行降维和特征提取后，图像进入全连接层，全连接层就是起到分类器的作用，通常使用的是softmax分类器，通过把前面的有用信息整合，加上激活函数的非线性映射，把卷积输出的二维特征图转化成一维向量，从而实现分类输出。vgg模型是一种卷积神经网络。vgg16就是把之前的vgg网络的1*1卷积核都换成3*3卷积核，从头到尾只有3*3卷积与2*2池化，增加网络深度的同时可以有效的提升网络训练的效果，相对于其他vgg模型，其对数据集的泛化能力更强。resnet50即残差神经网络，是指通过对传统卷积神经网络加入残差学习，使得深层网络里梯度扩散以及准确度降低的问题，使得网络层次可以越来越深，既使得精度提高，又加快了速度。densenet的核心思想是将每一层与后面的层连接，即对某一层而言，它的输入是前面所有层输出的连结。densenet减轻梯度消失，提高了特征的传播效率和利用效率，减少了网络的参数量
[0033]
首先本实施方式利用普通深度学习方法进行实验，深度学习的提出重点在于建立和模拟大脑在分析学习时的神经网络。深度学习有三大优点，一是可以处理大数据，二是计算能力强，三是不断创新的算法。
[0034]
选择vgg16、resnet50、densenet121三种网络模型分别对图像进行分类处理，将图像分为训练集、验证集、测试集，通过训练集确定神经网络训练模型的模型权重；通过验证集确定网络模型以及调整模型的超参数；测试集是检验模型的泛化能力。通过调参，实验的超参数选取为：
[0035]
(1)学习衰减率：0.001。
[0036]
(2)动量：0.5。
[0037]
(3)学习率：0.01。
[0038]
(4)权值衰减率：0.0005。
[0039]
(5)批量：32。
[0040]
(6)轮数：60。
[0041]
(7)训练算法：随机梯度下降。
[0042]
经过对竹片图像以及网络提取特征的研究和分析，以及经过对不同类别竹片图像的数据分布研究分析，发现：针对于竹片图像的分类而言，以上述超参数为基础进行模型超参数的调整，相比其他超参数作为训练基础的训练过程，可以缩短48％以上的超参数确定时间，进而极大地缩短训练时间，即本发明可以极大地缩短训练时间，同时以上述超参数为基础进行模型超参数的调整，还能够有效的缩短确定模型参数(训练过程确定的参数)的时间，并且保证确定模型参数对应的模型识别准确率。
[0043]
再利用迁移学习方法进行实验，迁移学习是把已学训练好的模型参数迁移到新的模型来帮助新模型训练。考虑到大部分数据或任务是存在相关性的，所以通过迁移学习，可以将已经学到的模型参数通过某种方式来分享给新模型从而加快并优化模型的学习效率。利用此方法，可以有效解决小数据样本过拟合等问题。为了跟前文普通深度学习方法对应起来，使用迁移学习方法的时候，同样是选择了vgg16、resnet50以及densenet121三种模型进行实验。
[0044]
通过混淆矩阵，来对上述得到的准确率进行对比。混淆矩阵是数据科学、数据分析和机器学习中总结分类模型预测结果的情形分析表，以矩阵形式将数据集中的记录按照真实的类别与分类模型作出的分类判断两个标准进行汇总。
[0045]
预测值为1，记为p；预测值为0，记为n；
[0046]
预测值与真实值相同，记为t；预测值与真实值相反，记为f；
[0047]
tp表示真阳，即真实为1，预测为1；fn表示假阴，即真实为1，预测为0；fp表示假阳，即真实为0，预测为1；tn表示真阴，即真实为0，预测为0。
[0048]
评判混淆矩阵的好坏，需要看4个指标，分别是准确率、精确率、召回率和f1-score。准确率＝(tp+tn)/(tp+fp+tn+fn)，精确率＝tp/(tp+fp)，召回率＝tp/(tp+fn)，经过对比以后，发现基于迁移学习的densenet121网络的实验结果最优，准确率达到98％，densenet121的混淆矩阵图如图2所示。。最后利用matlab软件画出6种神经网络实验结果的折线图，使实验结果对比更加明显直观，如图1所示。
[0049]
利用本发明的方法进行仿真，可以提升分类准确率以及减少人工劳动成本，通过实验，得到的最高分类准确率达到了98％，使得检测准确率得到了提升，并且该发明是利用计算机对程序进行操控，大大减少了人工筛检的时间、精力、金钱，节省了成本，同时一次性分类图像可以达到成千上万张，节省时间。最终效果最好的神经网络分类识别准确率达到98％以上，能很好的完成竹片筛检分类的任务，并且耗时短，6000多张图片只需要40分钟左右就可全部识别完毕，大大减少了人工成本的消耗。并且该发明是利用计算机对程序进行操控，大大减少了人工筛检的时间、精力、金钱，节省了成本，同时一次性分类图像可以达到成千上万张，节省时间。
[0050]
本发明还可有其它多种实施例，在不背离本发明精神及其实质的情况下，本领域技术人员当可根据本发明作出各种相应的改变和变形，但这些相应的改变和变形都应属于
本发明所附的权利要求的保护范围。