(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210686123.X (22)申请日 2022.06.16 (71)申请人 厦门稿定股份有限公司 地址 361006 福建省厦门市思明区东浦路 浦南一路10号第三层G02 申请人 稿定 （厦门） 科技有限公司 (72)发明人 曾志先　靳鸿蔚　 (74)专利代理 机构 北京慧加伦知识产权代理有 限公司 16 035 专利代理师 李强 (51)Int.Cl. G06T 11/00(2006.01) G06T 11/40(2006.01) G06V 10/762(2022.01) (54)发明名称 配色方案生 成模型训练方法、 配 色方案生成 方法及装置 (57)摘要 本公开涉及一种配色方案生成模型训练方 法、 配色方案生成方法及装置。 该训练方法包括： 获取图片数据集； 通过颜色聚类方法提取所述图 片数据集中的每张图片的第一主色信息； 根据每 张图片的第一主色信息， 形成每张图片各自的输 入数据和标签数据； 利用全部所述输入数据和所 述标签数据对 所述配色方案生 成模型进行训练。 通过本公开的方案， 可以通过图片获得模型的输 入数据和标签数据， 并且利用输入 数据和标签数 据对模型进行训练， 从而可以训练出能够生成与 图片相适配的配 色方案的配色方案生成模型,进 而可以根据图片通过简单快速的方式得到适配 的配色方案 。 权利要求书2页 说明书9页 附图3页 CN 115147505 A 2022.10.04 CN 115147505 A 1.一种配色方案生成模型训练方法， 其中， 所述方法包括： 获取图片数据集； 通过颜色聚类方法提取 所述图片数据集中的每张图片的第一主色信息； 根据每张图片的第一主色信息， 形成每张图片各自的输入数据和标签数据； 利用全部所述输入数据和所述标签数据对所述配色方案生成模型进行训练。 2.根据权利要求1所述的配色方案生成模型训练方法， 其中， 所述颜色聚类方法包括K ‑ means聚类算法。 3.根据权利要求1所述的配色方案生成模型训练方法， 其中， 所述第 一主色信 息包括多 个主色各自的RGB值以及所述多个主色各自的占比， 根据每张图片的第一主色信息， 形成每 张图片各自的输入数据和标签数据包括： 根据所述多个主色各自的占比， 对所述多个主色按照占比从高到低的顺序进行排序； 根据所述排序， 将所述多个主色的RGB值组成数组， 并将所述数组作为所述输入数据， 其中每个主色的RGB值作为所述数组的一个元 素； 将所述输入数据中的第一个元素删除， 并将随后所有元素向前移动一位， 并在最后一 个元素后面加入一个预设元 素， 以形成所述标签数据。 4.根据权利要求1所述的配色方案生成模型训练方法， 其中， 利用全部所述输入数据和 所述标签数据对所述配色方案生成模型进行训练包括： 将所述输入数据输入所述配色方案生成模型， 获得输出 结果； 根据所述标签数据和所述输出结果， 通过欧式距离计算方法获得损失值作为迭代模型 的参数， 其中所述标签数据中的所述预设元素中的数据与所述输出结果中的最后一个元素 的数据不进行欧式距离计算； 根据所述损失值， 通过Adam算法优化所述配色方案生成模型。 5.一种配色方案生成方法， 其中， 所述方法包括： 获取生成配色方案所用的图片； 根据所述图片， 获取第二主色信息； 基于所述第 二主色信 息， 通过配色方案生成模型， 获得用于所述图片的配色方案， 其中 所述配色方案生成模型通过权利要求1至4中任一项所述的配色方案生成模型训练方法训 练得到， 并且其中所述第二主色信息中的主色数量小于所述第一主色信息中的主色数量。 6.根据权利要求5所述的配色方案生成方法， 其中， 基于所述第二主色信息， 通过配色 方案生成模型， 获得用于所述图片的配色方案包括： 基于所述第 二主色信 息， 通过配色方案生成模型， 获得输出数据， 其中所述输出数据的 最后一行 数据为配色数据； 根据预设配色数量和所述配色数据， 通过配色方案生成模型获得用于所述图片的配色 方案。 7.一种配色方案生成模型训练装置， 其中， 所述装置包括： 第一获取模块， 其用于获取图片数据集； 主色提取模块， 其用于通过颜色聚类方法提取所述图片数据集中的每张图片的主色信 息； 数据生成模块， 其用于根据每张图片的主色信息， 形成每张图片各自的输入数据和标权　利　要　求　书 1/2 页 2 CN 115147505 A 2签数据； 训练模块， 其用于利用全部所述输入数据和所述标签数据对所述配色方案生成模型进 行训练。 8.一种配色方案生成装置， 其中， 所述装置包括： 第二获取模块， 其用于获取生成配色方案所用的图片； 第三获取模块， 其用于根据所述图片， 获取主色信息； 配色方案生成模块， 其用于基于所述主色信 息， 通过配色方案生成模型， 获得用于所述 图片的配色方案， 其中所述配色方案生成模型通过权利要求7所述的配色方案生成模型训 练装置训练得到 。 9.一种电子设备， 其中， 所述电子设备包括存储器和处理器， 所述存储器中存储有计算 机程序， 所述处理器执行所述计算机程序时， 实现如权利要求1至4中任一项所述的配色方 案生成模型训练方法， 或者实现如权利要求5或6所述的配色方案生成方法。 10.一种计算机可读存储介质， 其中， 所述存储介质存储有计算机程序， 所述计算机程 序被执行时， 实现如权利要求1至4中任一项所述的配色方案生成模型训练方法， 或者实现 如权利要求5或6所述的配色方案生成方法。权　利　要　求　书 2/2 页 3 CN 115147505 A 3