(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202211020688.0 (22)申请日 2022.08.24 (71)申请人 北京奇艺世纪科技有限公司 地址 100080 北京市海淀区海淀北一 街2号 鸿城拓展大厦10、 1 1层 (72)发明人 彭瑶　王亦梁　韩直彬　 (74)专利代理 机构 北京柏杉松知识产权代理事 务所(普通 合伙) 11413 专利代理师 李冬　孟维娜 (51)Int.Cl. G06V 10/56(2022.01) G06V 10/762(2022.01) G06V 10/774(2022.01) G06V 10/82(2022.01) (54)发明名称 一种图片的主题色确定方法、 模型训练方法 及装置 (57)摘要 本发明实施例提供了一种图片的主题色确 定方法、 模型训练方法及装置， 根据待处理图片 包含的各待处理颜色所属的色系， 对 各待处理颜 色进行分组， 得到多个待处理颜色分组； 针对每 一待处理颜色分组， 基于该待处理颜色分组中的 各待处理颜色， 计算该待处理颜色分组对应的待 处理颜色差异值和待处理数目均值； 将待处理颜 色差异值和待处理数目均值输入至预先训练的 目标颜色预测模 型， 得到该待处理颜色分组的待 处理置信度； 从对应的待处理置信度大于第一阈 值的各待处理颜色分组中， 选择一个目标颜色分 组； 根据目标颜色分组中的各待处理颜色对应的 像素值， 计算待处理图片的主题色。 基于此， 可以 降低人工成本和时间成本， 进而提高确定图片的 主题色的效率。 权利要求书3页 说明书21页 附图8页 CN 115424039 A 2022.12.02 CN 115424039 A 1.一种图片的主题色确定方法， 其特 征在于， 所述方法包括： 根据待处理图片包含的各待处理颜色所属的色系， 对待处理图片包含的各待处理颜色 进行分组， 得到多个颜色 分组， 作为待处 理颜色分组； 针对每一待处理颜色分组， 基于该待处理颜色分组中的各待处理颜色， 计算该待处理 颜色分组对应的待处理颜色差异 值和待处理数目均值； 其中， 所述待处理颜色差异 值表示： 该待处理颜色分组中各待处理颜色的离散程度； 所述待处理数目均值表示： 该待处理颜色 分组中的待处 理颜色在所述待处 理图片中的占比； 将所述待处理颜色差异值和所述待处理数目均值输入至预先训练的目标颜色预测模 型， 得到所述目标颜色预测模型输出的该待处 理颜色分组的置信度， 作为待处 理置信度； 从对应的待处理置信度大于第一阈值的各待处理颜色分组中， 选择一个颜色分组， 作 为目标颜色 分组； 根据所述目标颜色分组中的各待处理颜色对应的像素值， 计算所述待处理图片的主题 色。 2.根据权利要求1所述的方法， 其特征在于， 所述针对每一待处理颜色分组， 基于该待 处理颜色分组中的各待处理颜色， 计算该待处理颜色分组对应的待处理颜色差异值和待处 理数目均值， 包括： 针对每一待处理颜色分组， 按照该待处理颜色分组中的各待处理颜色对应的第 一数目 从大至小的排列顺序， 计算该待处理颜色分组中每相 邻的两个待处理颜色之间的颜色差异 值， 作为第一颜色差异 值； 计算该待处理颜色分组对应的各第一颜色差异值的平均值， 得到 该待处理颜色分组对应的待处理颜色差异 值； 其中， 一个待处理颜色对应的第一数目为： 所 述待处理图片中包 含该待处 理颜色的像素点的数目； 计算该待处理颜色分组中的各待处理颜色对应的第 一数目的平均值， 得到该待处理颜 色分组对应的待处 理数目均值。 3.根据权利要求1所述的方法， 其特征在于， 所述根据待处理图片包含的各待处理颜色 所属的色系， 对待处理图片包含的各待处理颜色进 行分组， 得到多个颜色分组， 作为待处理 颜色分组， 包括： 基于待处理图片中的各待处理颜色所属的色系， 对各待处理颜色进行聚类， 得到多个 颜色分组， 作为待处理颜色分组； 其中， 每两个待处理颜色分组的聚类中心之间的第二颜色 差异值大于第二阈值； 每一待处理颜色与所属待处理颜色分组的聚类中心之间的第二颜色 差异值， 小于该待处 理颜色与其 他待处理颜色分组的聚类中心之间的第二颜色差异值。 4.根据权利要求3所述的方法， 其特征在于， 在所述根据待处理图片包含的各待处理颜 色的所属的色系， 对待处理图片包含的各待处理颜色进 行分组， 得到多个颜色分组， 作为待 处理颜色分组之前， 所述方法还 包括： 按照待处理图片包含的颜色对应的第 一数目从大到小的排列顺序， 从待处理图片包含 的颜色中， 确定前第二数目个颜色， 作为待处 理颜色。 5.根据权利要求1所述的方法， 其特征在于， 在所述根据待处理图片包含的各待处理颜 色的所属的色系， 对待处理图片包含的各待处理颜色进 行分组， 得到多个颜色分组， 作为待 处理颜色分组之前， 所述方法还 包括： 针对待处理图片中的每一像素点， 如果该像素点的颜色的亮度值属于预设亮度区间，权　利　要　求　书 1/3 页 2 CN 115424039 A 2确定该像素点的颜色为备选颜色； 从各备选颜色中， 确定对应的第 一数目大于第 三阈值的备选颜色， 作为待处理颜色； 其 中， 一个备选颜色对应的第一数目为： 所述待处 理图片中包 含该备选颜色的像素点的数目。 6.根据权利要求5所述的方法， 其特征在于， 所述针对待处理图片中的每一像素点， 如 果该像素点的颜色的亮度值属于预设亮度区间， 确定该像素点的颜色为备选颜色， 包括： 针对待处理图片中的每一像素点， 如果该像素点的颜色的亮度值属于第 一预设亮度区 间， 确定该像素点的颜色为备选颜色； 或者， 针对待处理图片中的每一像素点， 如果该像素点的颜色的亮度值属于第 二预设亮度区 间， 且该颜色的饱和度值不属于预设饱和度区间， 确定该像素点的颜色为备选颜色； 其中， 所述第二预设亮度区间属于所述第一预设亮度区间。 7.根据权利要求6所述的方法， 其特征在于， 在所述针对待处理图片中的每一像素点， 如果该像素点的颜色的亮度值属于预设亮度区间， 确定该像素点的颜色为备选颜色之前， 所述方法还 包括： 获取原始图片， 并提取 所述原始图片中除指定对象以外的图像区域， 得到待处 理图片； 针对所述待处理图片中的每一像素点， 基于该像素点的像素值， 计算该像素点的颜色 的亮度值和饱和度值。 8.根据权利要求1至7任一项所述的方法， 其特征在于， 在所述根据所述目标颜色分组 中的各待处 理颜色对应的像素值， 得到所述待处 理图片的主题色之后， 所述方法还 包括： 将所述待处理图片的主题色， 作为所述待处理图片所属的原始图片的主题色， 并存储 所述原始图片的主题色； 在接收到客户端发送的针对所述原始图片的获取请求后， 向所述客户端发送所述原始 图片和所述原始图片的主题色， 以使 所述客户端在接收到所述原始图片和所述原始图片的 主题色后， 在所述客户端的显示界面中显示所述原始图片， 并将所述显示界面中除所述原 始图片外的其 他区域的颜色 设置为所述原 始图片的主题色。 9.一种模型训练方法， 其特 征在于， 所述方法包括： 根据样本图片包含的各样本颜色所属的色系， 对所述样本图片包含的各样本颜色进行 分组， 得到多个颜色 分组， 作为样本颜色 分组； 基于所述样本颜色分组中的各样本颜色， 计算所述样本颜色分组对应的样本颜色差异 值和样本数目均值； 其中， 所述样 本颜色差异 值表示： 所述样本颜色分组中的各样本颜色的 离散程度； 所述样本数目均值表示： 所述样本颜色分组中的样本颜色在所述样本图片 中的 占比； 获取所述样本颜色分组的置信度， 作为样本置信度； 其中， 所述样本置信度表示： 所述 样本颜色 分组中的样本颜色能够表征 所述样本图片的整体颜色特 征的概率； 将所述样本颜色差异值、 所述样本数目均值输入至初始结构的颜色预测模型， 得到所 述初始结构的颜色预测模型输出的所述样本颜色 分组的置信度， 作为预测置信度； 计算表示所述预测置信度和所述样本 置信度之间的差异的损失函数值； 基于计算得到的损失函数值对初始结构的颜色预测模型的模型参数进行调 整， 直至达 到预设的收敛 条件， 得到训练好的所述目标颜色预测模型。权　利　要　求　书 2/3 页 3 CN 115424039 A 3