发明公开
- 专利标题: 颜色识别模型训练方法、颜色识别方法、设备及存储介质
-
申请号: CN202411049490.4申请日: 2024-08-01
-
公开(公告)号: CN118570589A公开(公告)日: 2024-08-30
- 发明人: 段富治 , 施夏梦 , 蔡丹枫 , 余言勋 , 王亚运 , 黄宇 , 牛中彬 , 郝行猛
- 申请人: 浙江大华技术股份有限公司
- 申请人地址: 浙江省杭州市滨江区滨安路1187号
- 专利权人: 浙江大华技术股份有限公司
- 当前专利权人: 浙江大华技术股份有限公司
- 当前专利权人地址: 浙江省杭州市滨江区滨安路1187号
- 代理机构: 深圳市威世博知识产权代理事务所
- 代理商 严翠霞
- 主分类号: G06V10/774
- IPC分类号: G06V10/774 ; G06V10/56 ; G06V10/764
摘要:
本申请公开了一种颜色识别模型训练方法、颜色识别方法、设备及存储介质,该颜色识别模型训练方法包括:将待识别样本图像和待识别样本图像的样本标签输入待训练模型中进行模型训练;基于待训练模型对待识别样本图像进行多通道特征提取处理得到多通道特征,以及对待识别样本图像进行颜色识别处理得到目标颜色类别;基于多通道特征进行对比损失计算,得到对比损失,以及基于目标颜色类别和样本标签进行分类损失计算,得到分类损失;构建对比损失和分类损失的联合损失;基于每次模型训练的联合损失调整待训练模型的模型参数直至联合损失达到预设标准,得到训练好的颜色识别模型。上述方案,能够提高颜色识别准确率。