颜色识别模型训练方法、颜色识别方法、设备及存储介质
摘要:
本申请公开了一种颜色识别模型训练方法、颜色识别方法、设备及存储介质,该颜色识别模型训练方法包括:将待识别样本图像和待识别样本图像的样本标签输入待训练模型中进行模型训练;基于待训练模型对待识别样本图像进行多通道特征提取处理得到多通道特征,以及对待识别样本图像进行颜色识别处理得到目标颜色类别;基于多通道特征进行对比损失计算,得到对比损失,以及基于目标颜色类别和样本标签进行分类损失计算,得到分类损失;构建对比损失和分类损失的联合损失;基于每次模型训练的联合损失调整待训练模型的模型参数直至联合损失达到预设标准,得到训练好的颜色识别模型。上述方案,能够提高颜色识别准确率。
0/0