一种基于深度学习的颜料光谱辨识方法转让专利

申请号 : CN201911337833.6

文献号 : CN111191701B

文献日 :

基本信息:

PDF:

法律信息:

相似专利:

发明人 : 范尧张朋昌唐兴佳王爽于涛高晓惠卢孝强胡炳樑

申请人 : 中国科学院西安光学精密机械研究所

摘要 :

本发明提供了一种基于深度学习的颜料光谱辨识方法,解决现有利用颜料光谱来识别颜料种类的方法,存在识别准确率低的问题。该方法包括以下步骤:1)采集颜料块的高光谱数据,获得高光谱数据块;2)去除高光谱数据块中成像质量差的高光谱数据,并对高光谱数据进行降噪处理;3)对采集的高光谱图像数据块进行分割得到若干图像块,将每个图像块的所有像素光谱曲线排成矩阵,对每个矩阵的每一条光谱曲线样本添加颜料标签,将每个矩阵的所有光谱曲线样本分为测验单元,4)建立深度学习神经网络模型,5)训练神经网络模型,调整网络模型的参数,6)使用调整好参数的神经网络模型进行测试,并统计其测试结果,7)计算及调整。

权利要求 :

1.一种基于深度学习的颜料光谱辨识方法,其特征在于,包括以下步骤:

1)采集

采集颜料块的高光谱数据,获得高光谱数据块;

2)预处理

去除高光谱数据块中成像质量差的高光谱数据,并对高光谱数据进行降噪处理;

3)分割并添加标签

3.1)对采集的高光谱图像数据块进行分割,得到若干a*a*b图像块,所述a、b均大于等于0;

3.2)将每个a*a*b图像块的所有像素光谱曲线排成c*b的矩阵,所述c=a*a;

其中,矩阵的每列表示一条光谱曲线样本,每行表示光谱样本序号;

3.3)对每个矩阵的每一条光谱曲线样本添加颜料标签,且每个矩阵的所有颜料标签属于同一种颜料;

3.4)将每个矩阵的所有光谱曲线样本分为测验单元,所述测验单元包括训练集、测试集,且训练集的数量大于测试集;

4)建立深度学习神经网络模型

建立深度学习神经网络模型,其结构包括6层,分别为输入卷积层、第一卷积层、第二卷积层、第三卷积层、第四卷积层、输出分类层;

所述输入卷积层的输出通道与第一卷积层的输入通道相同;

所述第一卷积层的输出通道与第二卷积层的输入通道相同;

所述第二卷积层的输出通道与第三卷积层的输入通道相同;

所述第三卷积层的输出通道与第四卷积层的输入通道相同;

所述第四卷积层的输出通道与输出分类层的输入通道相同;

所述第一卷积层、第二卷积层、第三卷积层、第四卷积层的节点依次呈指数下降;

5)训练神经网络模型,调整网络模型的参数将训练集作为样本输入神经网络模型中进行训练,调整网络模型的参数,所述参数包括权重W和偏置B,并保留调整好的参数;

6)测试

使用调整好参数的神经网络模型在测试集上进行测试,并统计其测试结果;

7)计算及调整

计算测试结果的正确率,若正确率高于所给定阈值,则分类结果准确;

若正确率低于所给定阈值,返回步骤5),重新调整网络模型的条件参数,直至分类结果准确;

所述条件参数包括学习率、批次大小、优化方法、迭代次数、激活函数。

2.根据权利要求1所述基于深度学习的颜料光谱辨识方法,其特征在于:步骤3)中,所述测验单元还包括验证集;

在步骤5)和步骤6)之间还包括步骤A)验证:将验证集作为样本输入神经网络模型,在验证集上验证神经网络模型的准确性。

3.根据权利要求2所述基于深度学习的颜料光谱辨识方法,其特征在于:步骤3)中,所述图像块的规格为80*80*520,矩阵的规格为6400*520。

4.根据权利要求3所述基于深度学习的颜料光谱辨识方法,其特征在于:步骤4)中,所述输入卷积层输入520个节点,输出512个节点;

所述第一卷积层输入512个节点,输出256个节点;

所述第二卷积层输入256个节点,输出128个节点;

所述第三卷积层输入128个节点,输出64个节点;

所述第四卷积层输入64个节点,输出32个节点;

所述输出分类层输入32个节点,输出30个节点。

5.根据权利要求4所述基于深度学习的颜料光谱辨识方法,其特征在于:步骤3.4)中,所述训练集、测试集、验证集的比例分别为70%、15%、15%。

6.根据权利要求5所述基于深度学习的颜料光谱辨识方法,其特征在于:步骤7)中,所述给定阈值为80%。

说明书 :

一种基于深度学习的颜料光谱辨识方法

技术领域

[0001] 本发明涉及颜料辨识技术,尤其涉及一种基于深度学习的颜料光谱辨识方法。

背景技术

[0002] 在进行颜料物质识别时,现有一些识别方法需要采集颜料样本,对颜料物质是有损或微损的,如采用X光衍射方法、拉曼光谱分析方法、电镜扫描分析方法等。
[0003] 现有利用颜料光谱来识别颜料种类是一类无损的可行方法,利用测量光谱与已知光谱进行比较,可以得出颜料类别,常用方法有光谱角填图、光谱相关测量等,但由于不同类别颜料光谱的区分度不大,因此识别准确率不高。

发明内容

[0004] 为了解决现有利用颜料光谱来识别颜料种类的方法,存在识别准确率低的技术问题,本发明提供了一种基于深度学习的颜料光谱辨识方法。
[0005] 为实现上述目的,本发明提供的技术方案是:
[0006] 一种基于深度学习的颜料光谱辨识方法,其特殊之处在于,包括以下步骤:
[0007] 1)采集
[0008] 采集颜料块的高光谱数据,获得高光谱数据块;
[0009] 2)预处理
[0010] 去除高光谱数据块中成像质量差的高光谱数据,并对高光谱数据进行降噪处理;
[0011] 3)分割并添加标签
[0012] 3.1)对采集的高光谱图像数据块进行分割,得到若干a*a*b图像块,所述a、b均大于等于0;
[0013] 3.2)将每个a*a*b图像块的所有像素光谱曲线排成c*b的矩阵,所述c=a*a;
[0014] 其中,矩阵的每列表示一条光谱曲线样本,每行表示光谱样本序号;
[0015] 3.3)对每个矩阵的每一条光谱曲线样本添加颜料标签,且每个矩阵的所有颜料标签属于同一种颜料;
[0016] 3.4)将每个矩阵的所有光谱曲线样本分为测验单元,所述测验单元包括训练集、测试集,且训练集的数量大于测试集;
[0017] 4)建立深度学习神经网络模型
[0018] 建立深度学习神经网络模型,其结构包括6层,分别为输入卷积层、第一卷积层、第二卷积层、第三卷积层、第四卷积层、输出分类层;
[0019] 所述输入卷积层的输出通道与第一卷积层的输入通道相同;
[0020] 所述第一卷积层的输出通道与第二卷积层的输入通道相同;
[0021] 所述第二卷积层的输出通道与第三卷积层的输入通道相同;
[0022] 所述第三卷积层的输出通道与第四卷积层的输入通道相同;
[0023] 所述第四卷积层的输出通道与输出分类层的输入通道相同;
[0024] 所述第一卷积层、第二卷积层、第三卷积层、第四卷积层的节点依次呈指数下降;
[0025] 5)训练神经网络模型,调整网络模型的参数
[0026] 将训练集作为样本输入神经网络模型中进行训练,调整网络模型的参数,所述参数包括权重W和偏置B,并保留调整好的参数;
[0027] 6)测试
[0028] 使用调整好参数的神经网络模型在测试集上进行测试,并统计其测试结果;
[0029] 7)计算及调整
[0030] 计算测试结果的正确率,若正确率高于所给定阈值,则分类结果准确;
[0031] 若正确率低于所给定阈值,返回步骤5),重新调整网络模型的条件参数,直至分类结果准确;
[0032] 所述条件参数包括学习率、批次大小、优化方法、迭代次数、激活函数。
[0033] 进一步地,步骤3)中,所述测验单元还包括验证集;
[0034] 在步骤5)和步骤6)之间还包括步骤A)验证:
[0035] 将验证集作为样本输入神经网络模型,在验证集上验证神经网络模型的准确性。
[0036] 进一步地,步骤3)中,所述图像块的规格为80*80*520,矩阵的规格为6400*520。
[0037] 进一步地,步骤4)中,所述输入卷积层输入520个节点,输出512个节点;
[0038] 所述第一卷积层输入512个节点,输出256个节点;
[0039] 所述第二卷积层输入256个节点,输出128个节点;
[0040] 所述第三卷积层输入128个节点,输出64个节点;
[0041] 所述第四卷积层输入64个节点,输出32个节点;
[0042] 所述输出分类层输入32个节点,输出30个节点。
[0043] 进一步地,步骤3.4)中,所述训练集、测试集、验证集的比例分别为70%、15%、15%。
[0044] 进一步地,步骤7)中,所述给定阈值为80%。
[0045] 与现有技术相比,本发明的优点是:
[0046] 1、本发明颜料光谱辨识方法通过对颜料块进行高光谱数据采集,并在对高光谱数据预处理的基础上建立神经网络模型,使用调整好参数的网络模型对颜料种类进行识别,准确得出颜料种类,可广泛应用于高光谱数据处理领域,并且对于其他物体种类辨识技术具有良好的借鉴性。
[0047] 2、本发明颜料光谱辨识方法在测试前,可在验证集上验证神经网络模型的准确性,提高辨识的辨别精度。
[0048] 3、本发明颜料光谱辨识方法所使用的神经网络模型结构简单,且该模型的节点数量呈指数下降,能有效减小运算复杂度,提高运算效率,进而提高辨别效率。

附图说明

[0049] 图1是本发明基于深度学习的颜料光谱辨识方法的流程图;
[0050] 图2是本发明基于深度学习的颜料光谱辨识方法中深度学习神经网络模型的结构与运算过程示意图。

具体实施方式

[0051] 以下结合附图和具体实施例对本发明的内容作进一步详细描述。
[0052] 如图1所示,一种基于深度学习的颜料光谱辨识方法,对颜料块进行高光谱数据采集,并预处理的基础上建立卷积神经网络模型,对颜料种类进行识别,可广泛应用于高光谱数据处理领域,并且对于其他物体种类辨识技术具有良好的借鉴性。该方法包括以下步骤:
[0053] 1)采集
[0054] 利用高光谱相机Gaia Field采集颜料块的高光谱数据,获得高光谱数据块;
[0055] 2)预处理
[0056] 去除高光谱数据块中成像质量差的高光谱数据,并对高光谱数据进行降噪处理;
[0057] 3)分割并添加标签
[0058] 3.1)对采集的高光谱图像数据块进行分割,将其沿一个面分为若干80*80*520的小方块,得到若干80*80*520图像块;
[0059] 3.2)将每个80*80*520图像块的所有像素光谱曲线排成6400*520的矩阵,[0060] 其中,排成矩阵的每列表示一条光谱曲线样本(350nm‑1032nm),每一行表示光谱样本序号;
[0061] 3.3)对每个矩阵的每一条光谱曲线样本添加颜料标签,添加名称标签是针对每一条光谱曲线样本加一个标签,因为每个单独图像块表示的是同一种颜料,因此每个矩阵的所有颜料标签属于同一种颜料(令每个颜料名称标签属于同一区域);
[0062] 3.4)将每个矩阵的所有光谱曲线样本分为训练集、测试集、验证集,训练集、测试集、验证集分别占比为70%、15%、15%;
[0063] 4)建立深度学习神经网络模型
[0064] 建立深度学习神经网络模型,其结构包括6层,如图2所示,分别为输入卷积层(输入520、输出512)、第一卷积层(输入512、输出256)、第二卷积层(输入256、输出128)、第三卷积层(输入128、输出64)、第四卷积层(输入64、输出32)、输出分类层(输出30);
[0065] 输入卷积层输入为520个节点,对应的是一条样本谱线的520的像素(数据点),输出为中间值512个像素(数据点);
[0066] 第一卷积层输入为512个节点,输出为中间值256个像素(数据点);
[0067] 第二卷积层输入为256个节点,输出为中间值128个像素(数据点);
[0068] 第三卷积层输入为128个节点,输出为中间值64个像素(数据点);
[0069] 第四卷积层输入为64个节点,输出为中间值32个像素(数据点);
[0070] 输出分类层输入为32个节点,输出为30个像素(数据点)。
[0071] 深度学习神经网络模型的节点数量呈指数下降,能有效减小运算复杂度,提高运算效率,该模型结构简单,轻量,并有对称性,模型的节点数量可以自适应调整;
[0072] 5)训练神经网络模型,调整网络模型的参数
[0073] 将训练集作为样本输入神经网络模型中进行训练,调整网络模型的参数,所述参数包括权重W和偏置B,并保留调整好的权重W和偏置B;
[0074] 6)验证
[0075] 将验证集作为样本输入调整好参数的神经网络模型,在验证集上验证神经网络模型的准确性;
[0076] 7)测试
[0077] 使用验证后的神经网络模型在测试集上进行测试,并统计其测试结果;
[0078] 8)计算及调整
[0079] 计算测试结果的正确率,正确率为统计正确样本所占总样本数的比例,若正确率高于所给定阈值,通常正确率的给定阈值为80%,则分类结果准确并停止;其中,输出层使用softmax函数,根据最大概率准则,输出最大值所在的标签类即为预测该样本的分类,已经训练好的神经网络针对每一个测试样本都会有一个测试分类结果,比方说测试集有5000个样本,这5000个样本本来是有标签的(以其中一种颜料铅丹为例),根据设定阈值为80%计算,结果大于4000条测试分类结果是铅丹,那么就认为分类结果准确;
[0080] 若统计正确率低于所给定阈值,返回步骤5),则重新调整网络模型的条件参数,这里可以调整的条件参数有学习率learing  rate、批次大小batch_size、优化方法optimizer、迭代次数、激活函数,直到正确率高于给定阈值,分类结果为止。
[0081] 或者,计算测试结果的错误率,错误率为统计错误样本所占总样本数的比例,若错误率低于所给定阈值,通常错误率的给定阈值为20%,则分类结果准确并停止;若错误率高于所给定阈值,则重新调整网络模型的条件参数,直到正确率低于给定阈值分类结果为止。
[0082] 该方法在绘画颜料辨识领域具有极大的价值,能无损识别颜料。适合工程的普遍应用,稳定性好。
[0083] 以上仅是对本发明的优选实施方式进行了描述,并不将本发明的技术方案限制于此,本领域技术人员在本发明主要技术构思的基础上所作的任何公知变形都属于本发明所要保护的技术范畴。