[发明专利]对象颜色的识别方法、装置、计算机设备及存储介质

权利要求书

1.一种对象颜色的识别方法，其特征在于，所述方法包括：

获取待识别对象的图像；

将所述待识别对象的图像输入至预先训练的颜色分类模型中，输出所述待识别对象的图像包含的各局部图像区域对应的候选颜色；

针对每个局部图像区域，如果该局部图像区域对应的候选颜色的数目为多个，则根据该局部图像区域对应的多个候选颜色和预设的颜色确定规则，确定该局部图像区域对应的目标颜色；

如果该局部图像区域对应的候选颜色的数目为一种，则将该局部图像区域对应的候选颜色，确定为该局部图像区域对应的目标颜色。

2.根据权利要求1所述的方法，其特征在于，所述将所述待识别对象的图像输入至预先训练的颜色分类模型中，输出所述待识别对象的图像包含的各局部图像区域对应的候选颜色，包括：

将所述待识别对象的图像输入至预先训练的颜色分类模型中，输出所述待识别对象的图像包含的各局部图像区域对应的各预设颜色的概率；

针对每个局部图像区域，在所述各预设颜色中，将概率大于预设概率阈值的预设颜色，确定该局部图像区域对应的候选颜色；或者，

针对每个局部图像区域，在所述各预设颜色中，按照概率由大到小的顺序，确定该局部图像区域对应的预设数目个候选颜色。

3.根据权利要求2所述的方法，其特征在于，所述各预设颜色包括无色调颜色或有色调颜色中的一种或多种。

4.根据权利要求1或2所述的方法，其特征在于，该局部图像区域对应的候选颜色的数目为两个，所述根据该局部图像区域对应的多个候选颜色和预设的颜色确定规则，确定该局部图像区域对应的目标颜色，包括：

如果该局部图像区域对应的两个候选颜色为对比色或互补色，则将概率最大的候选颜色，确定为该局部图像区域对应的目标颜色，所述候选颜色的概率是由所述预先训练的颜色分类模型输出的；

如果该局部图像区域对应的两个候选颜色为邻近色或同类色，则将该局部图像区域对应的两个候选颜色的复合色，确定为该局部图像区域对应的目标颜色。

5.根据权利要求4所述的方法，其特征在于，

当所述两个候选颜色均为无色调颜色时，所述两个候选颜色的复合色为灰度值在所述两个候选颜色的灰度值之间的灰色；

当所述两个候选颜色均为有色调颜色时，所述两个候选颜色的复合色为所述两个候选颜色的混合颜色；

当所述两个候选颜色分别为无色调颜色和有色调颜色时，所述两个候选颜色的复合色为所述两种候选颜色的中间态颜色。

6.根据权利要求1至5任一项所述的方法，其特征在于，所述方法还包括：

获取预先存储的训练样本集合，所述训练样本集合中包括多个样本对象的图像以及每个样本对象的图像包含的各局部图像区域对应的至少一种样本颜色；

根据各样本对象的图像以及各样本对象包含的各局部图像区域对应的至少一种样本颜色对待训练的颜色分类模型进行训练，得到经过训练的颜色分类模型。

7.一种对象颜色的识别装置，其特征在于，所述装置包括：

第一获取模块，用于获取待识别对象的图像；

输出模块，用于将所述待识别对象的图像输入至预先训练的颜色分类模型中，输出所述待识别对象的图像包含的各局部图像区域对应的候选颜色；

第一确定模块，用于针对每个局部图像区域，如果该局部图像区域对应的候选颜色的数目为多个，则根据该局部图像区域对应的多个候选颜色和预设的颜色确定规则，确定该局部图像区域对应的目标颜色；

第二确定模块，用于如果该局部图像区域对应的候选颜色的数目为一种，则将该局部图像区域对应的候选颜色，确定为该局部图像区域对应的目标颜色。