[发明专利]一种图片的主题色确定方法、模型训练方法及装置

权利要求书

1.一种图片的主题色确定方法，其特征在于，所述方法包括：

根据待处理图片包含的各待处理颜色所属的色系，对待处理图片包含的各待处理颜色进行分组，得到多个颜色分组，作为待处理颜色分组；

针对每一待处理颜色分组，基于该待处理颜色分组中的各待处理颜色，计算该待处理颜色分组对应的待处理颜色差异值和待处理数目均值；其中，所述待处理颜色差异值表示：该待处理颜色分组中各待处理颜色的离散程度；所述待处理数目均值表示：该待处理颜色分组中的待处理颜色在所述待处理图片中的占比；

将所述待处理颜色差异值和所述待处理数目均值输入至预先训练的目标颜色预测模型，得到所述目标颜色预测模型输出的该待处理颜色分组的置信度，作为待处理置信度；

从对应的待处理置信度大于第一阈值的各待处理颜色分组中，选择一个颜色分组，作为目标颜色分组；

根据所述目标颜色分组中的各待处理颜色对应的像素值，计算所述待处理图片的主题色。

2.根据权利要求1所述的方法，其特征在于，所述针对每一待处理颜色分组，基于该待处理颜色分组中的各待处理颜色，计算该待处理颜色分组对应的待处理颜色差异值和待处理数目均值，包括：

针对每一待处理颜色分组，按照该待处理颜色分组中的各待处理颜色对应的第一数目从大至小的排列顺序，计算该待处理颜色分组中每相邻的两个待处理颜色之间的颜色差异值，作为第一颜色差异值；计算该待处理颜色分组对应的各第一颜色差异值的平均值，得到该待处理颜色分组对应的待处理颜色差异值；其中，一个待处理颜色对应的第一数目为：所述待处理图片中包含该待处理颜色的像素点的数目；

计算该待处理颜色分组中的各待处理颜色对应的第一数目的平均值，得到该待处理颜色分组对应的待处理数目均值。

3.根据权利要求1所述的方法，其特征在于，所述根据待处理图片包含的各待处理颜色所属的色系，对待处理图片包含的各待处理颜色进行分组，得到多个颜色分组，作为待处理颜色分组，包括：

基于待处理图片中的各待处理颜色所属的色系，对各待处理颜色进行聚类，得到多个颜色分组，作为待处理颜色分组；其中，每两个待处理颜色分组的聚类中心之间的第二颜色差异值大于第二阈值；每一待处理颜色与所属待处理颜色分组的聚类中心之间的第二颜色差异值，小于该待处理颜色与其他待处理颜色分组的聚类中心之间的第二颜色差异值。

4.根据权利要求3所述的方法，其特征在于，在所述根据待处理图片包含的各待处理颜色的所属的色系，对待处理图片包含的各待处理颜色进行分组，得到多个颜色分组，作为待处理颜色分组之前，所述方法还包括：

按照待处理图片包含的颜色对应的第一数目从大到小的排列顺序，从待处理图片包含的颜色中，确定前第二数目个颜色，作为待处理颜色。

5.根据权利要求1所述的方法，其特征在于，在所述根据待处理图片包含的各待处理颜色的所属的色系，对待处理图片包含的各待处理颜色进行分组，得到多个颜色分组，作为待处理颜色分组之前，所述方法还包括：

针对待处理图片中的每一像素点，如果该像素点的颜色的亮度值属于预设亮度区间，确定该像素点的颜色为备选颜色；

从各备选颜色中，确定对应的第一数目大于第三阈值的备选颜色，作为待处理颜色；其中，一个备选颜色对应的第一数目为：所述待处理图片中包含该备选颜色的像素点的数目。

6.根据权利要求5所述的方法，其特征在于，所述针对待处理图片中的每一像素点，如果该像素点的颜色的亮度值属于预设亮度区间，确定该像素点的颜色为备选颜色，包括：

针对待处理图片中的每一像素点，如果该像素点的颜色的亮度值属于第一预设亮度区间，确定该像素点的颜色为备选颜色；

或者，

针对待处理图片中的每一像素点，如果该像素点的颜色的亮度值属于第二预设亮度区间，且该颜色的饱和度值不属于预设饱和度区间，确定该像素点的颜色为备选颜色；其中，所述第二预设亮度区间属于所述第一预设亮度区间。