基于卷积和自注意机制神经网络的织物颜色识别方法技术

基于卷积和自注意机制神经网络的织物颜色识别方法，属于图像识别技术领域。本发明专利技术提出一种全新的基于卷积神经网络的颜色预测模型CSAM，卷积层和注意层按原则穿插叠加，很好地提高了模型的泛化能力、容量和效率；通过考虑原始输入图像与一组参考彩色图像之间的差异，并在色差域对输入图像进行RGB编码，提取织物图像中的颜色识别特征，然后采用一种集成式的卷积神经网络架构，最后通过卷积层和最大池化层的特征计算降维后通过softmax激活函数的密集层，对各个分类器的预测结果加权平均，对织物颜色进行有效的分类预测。本发明专利技术采用了全新的神经网络架构，显著地提高织物颜色识别的识别成功率和准确率，且拥有良好的可移植性和泛用性。泛用性。泛用性。

【技术实现步骤摘要】
基于卷积和自注意机制神经网络的织物颜色识别方法


[0001]本专利技术属于图像识别
，具体涉及基于卷积和自注意机制神经网络的织物颜色识别方法。

技术介绍

[0002]色彩是视觉传达的重要组成部分。它也是几个研究领域的相关主题，随着服装行业的发展，对款式的需求越来越多元化，市场上的面料颜色种类也越来越繁杂，传统的织物颜色识别方法已经难以适应织物销售行业的发展要求，颜色识别在纺织工业中非常重要。在这种情况下，颜色是最重要在制造商和客户之间传递的信息的之一。在纺织品背景下使用深度学习进行颜色识别的研究仍然很少，我们希望通过提出一种新的基于深度学习的织物颜色识别方法来对此做出贡献。
[0003]深度学习是近几年机器学习的热门领域，也是目前最接近人工智能这个目标的学习方法。以视觉感知为例，机器学习方法解决问题的过程可以概括为以下几个步骤：
[0004]层信息感知：根据问题的特点和数据的属性，选择合适的机器学习算法，并设置相关超参数。
[0005]预处理：对原始数据进行清洗、去除噪声、填补缺失值、归一化等预处理操作，以保证数据的可靠性。
[0006]特征抽取：从原始数据中提取出具有代表性的特征，用于训练模型，这些特征可能是数值、类别、文本或图像等。
[0007]特征筛选：根据数据的实际情况，通过特征工程等方法对提取的特征进行筛选，以减少特征数量、降低噪声影响、提高模型的泛化能力。
[0008]建模(预测、识别)：使用训练数据训练机器学习模型，并使用测试数据验证和评估模型的性能，最终应用模型进行预测或识别等任务，进而解决问题。
[0009]总之，机器学习方法解决问题时，需要经过数据预处理、特征提取、特征筛选等多个步骤，才能训练出效果良好的模型，并对新的数据进行预测或识别等操作。但是现有市面上的图像识别技术依旧使用着较为落后的神经网络模型，其模型还不够完善并且识别的准确率并不高，并且只能够相应的数据进行处理，并不能够采用注意力的机制进一步提高神经网络模型的学习能力。
[0010]机器学习的两个基本方面——泛化和模型容量中.卷积层由于其强的归纳偏差先验，往往具有更好的泛化和更快的收敛速度，而注意层具有更高的模型容量，可以受益于更大的数据集。本专利技术鉴于调查发现将卷积层和注意层结合起来可以获得更好的泛化和容量；此外，本专利技术提出的CSAM神经网络颜色预测模型不仅具有良好的可扩展性，而且收敛速度更快，从而提高了效率。当仅使用数据进行训练时，本专利技术提出的CSAM模型的准确率达到86.0％，在相似的计算资源和训练条件下，与市面上现有的.技术NFNet相当。
[0011]目前，进行织物颜色识别的方案主要有三种：第一种是通过专业的颜色识别人员对成分进行检测。第二种是通过手机拍照的方式，通过图像颜色匹配的方式，对织物颜色进
行识别。
[0012]这两种颜色识别的方法都存在着较大的弊端，通过专业的颜色检测人员来鉴别织物的颜色虽然识别成功率和准确率较高，但这对人员的技术和经验要求相对较高，目前此类专业人员缺失，地域分布极不均匀，专门聘请此类人员也会有较高的成本；手机拍照识别的方式受外界干扰和硬件水平的影响较大。在拍摄的过程中，收到环境光的影响较大并且仅仅通过手机摄像头的条件难以捕捉织物的细节特征，在匹配的过程也仅仅采用了过时的颜色匹配方式。最终匹配的准确率不高，并不能达到准确匹配织物颜色的目的。
[0013]目前市面上关于织物颜色识别的论文与专利，如Multi
‑
Label Clothing Color Recognition Based on Mobile Devices、Clothing Color Classification Using Statistical Background Model and Deep Features等，这些方法在实际应用中，也存在着一定的问题。例如，对于某些特殊材质和颜色的衣物，模型的准确性可能受到很大影响。此外，在复杂场景下，在衣物与背景颜色相近时，算法也容易产生误判。此外，这些模型需要大量标注数据进行训练，训练难度较大。
[0014]衣物颜色识别的准确性和实用性仍有待提高。针对这些问题，由此可见，专利技术一种新的织物颜色识别方法的市场前景非常广泛，也能精准的解决面料行业辨色难的痛点。

技术实现思路

[0015]针对现有技术中存在的上述问题，本专利技术的目的在于提供一种基于卷积和自注意机制神经网络的织物颜色识别方法，能够提高面料识别对于成分检测的准确识别率以及识别速率。
[0016]本专利技术提供如下技术方案：
[0017]基于卷积和自注意机制神经网络的织物颜色识别方法，包括以下步骤：
[0018]S1、使用智能终端获取织物图像信息(亮度、色温、饱和度等方面的数据，同时也包括了纹理、图案、光泽度等其他关键信息。)；
[0019]S2、结合卷积网络与Transformer注意力机制，构建深度学习模型CSAM；
[0020]S3、对获取的图像信息进行图像预处理；
[0021]S4、对图像进行差分拼接，为深度学习模型创建相应的输入；
[0022]S5、将步骤S4中创建的输入数据作为训练集对深度学习模型CSAM进行训练；
[0023]S6、输入测试数据通过训练完成的深度学习模型CSAM对织物颜色进行分类预测。
[0024]进一步的，所述步骤S1的具体过程如下：
[0025]将原始图像映射到差分域，并且在色差域对输入图像进行RGB编码，提取织物图像中的颜色识别特征。
[0026]进一步的，所述深度学习模型CSAM架构中，深度卷积和自注意力结合在一个计算模块内，将不同类别的计算模型垂直的堆叠在一起，形成一个完整的网络；深度卷积和自注意力表示为一个预先定义好的感受野内的数值加权和。
[0027]进一步的，所述深度学习模型CSAM的构建过程如下：
[0028]2.1、将深度卷积神经网络与自注意机制融合；具体过程如下：
[0029]2.1.1)将深度卷积核自注意力都表示为一个在预定义的感受野中所有值的加权和操作；
[0030]2.1.2)通过归一化得到最终输出；
[0031]2.1.3)在softmax归一化之前或者之后将全局的静态卷积核与attention注意力矩阵进行相加，将卷积和注意力有效结合；
[0032]2.2、将卷积和注意力有效结合后，堆叠搭建一个完整的网络，网络选用卷积层和自注意力层混合层叠结构。
[0033]进一步的，所述步骤2.1.1中，卷积为在一个预定义好的感受野中提取特征，具体表达式如下：
[0034][0035]其中，Yi和xi分别表示i位置的输入输出；
[0036]自注意机制表达式如下：
[0037][0038]其中，Q，K，V是输入的Query、Key和Value向量，softmax是softmax函数，dk是Query和Key的关联度；
本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.基于卷积和自注意机制神经网络的织物颜色识别方法，其特征在于包括以下步骤：S1、使用智能终端获取织物颜色图像信息；S2、结合卷积网络与Transformer自我注意力机制，构建颜色预测模型CSAM；S3、对获取的图像信息进行图像预处理；S4、对图像进行差分拼接，为深度学习模型创建相应的输入；S5、将步骤S4中创建的输入数据作为训练集对深度学习模型CSAM进行训练；S6、通过训练完成的颜色预测模型CSAM对织物颜色进行分类预测。2.根据权利要求1所述的基于卷积和自注意机制神经网络的织物颜色识别方法，其特征在于所述步骤S1的具体过程如下：将原始图像映射到差分域，并且在色差域对输入图像进行RGB编码，提取织物图像中的颜色识别特征。3.根据权利要求2所述的基于卷积和自注意机制神经网络的织物颜色识别方法，其特征在于所述颜色预测模型CSAM架构中，深度卷积和自注意力结合在一个计算模块内，将不同类别的计算模型垂直的堆叠在一起，形成一个完整的网络；深度卷积和自注意力表示为一个预先定义好的感受野内的数值加权和。4.根据权利要求3所述的基于卷积和自注意机制神经网络的织物颜色识别方法，其特征在于所述颜色预测深度学习模型CSAM的构建过程如下：2.1、将深度卷积神经网络与自注意机制融合；具体过程如下：2.1.1)将深度卷积核自注意力都表示为一个在预定义的感受野中所有值的加权和操作；2.1.2)通过归一化得到最终输出；2.1.3)在softmax归一化之前或者之后将全局的静态卷积核与attention注意力矩阵进行相加，将卷积和注意力有效结合；2.2、将卷积和注意力有效结合后，堆叠搭建一个完整的网络，网络包括卷积层和自注意力层混合层叠结构。5.根据权利要求4所述的基于卷积和自注意机制神经网络的织物颜色识别方法，其特征在...

【专利技术属性】
技术研发人员：邵一鸣，罗钇凯，邬臻林，俞可扬，沈启承，王郅祺，高义雄，林璐凝，胡杰，吴子朝，章国道，
申请(专利权)人：杭州电子科技大学，
类型：发明
国别省市：