(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210426094.3 (22)申请日 2022.04.21 (71)申请人 清华大学 地址 100084 北京市海淀区清华园 申请人 北京达佳互联信息技 术有限公司 (72)发明人 黄高　郭嘉懿　杜超群　黄慧娟　 万鹏飞　 (74)专利代理 机构 北京三高永信知识产权代理 有限责任公司 1 1138 专利代理师 李加欣 (51)Int.Cl. G06T 7/00(2017.01) G06T 3/00(2006.01) G06K 9/62(2022.01) G06V 10/774(2022.01)G06V 10/74(2022.01) (54)发明名称 图像检测模型的训练方法、 图像检测方法、 装置及设备 (57)摘要 本公开是关于一种图像检测模型的训练方 法、 图像检测方法、 装置及设备， 属于计算机技术 领域。 方法包括： 获取属于目标风格的标准图像； 调用正在训练的图像生成模型， 对原始图像进行 风格迁移得到目标图像； 基于目标图像确定与目 标图像的风格匹配的第一样本图像， 基于图像生 成模型的训练次数， 确定样本标签； 调用待训练 的图像检测模 型， 将第一样本图像与标准图像进 行对比， 得到第一检测标签； 基于样本标签和第 一检测标签， 训练待训练的图像检测模型， 得到 训练完成的图像检测模型。 本公开提供了一种自 动生成样本图像和样本标签的方式， 然后利用样 本图像、 样本标签和标准图像， 训练图像检测模 型， 从而使用图像检测模型来对图像风格进行检 测。 权利要求书2页 说明书22页 附图6页 CN 114998196 A 2022.09.02 CN 114998196 A 1.一种图像 检测模型的训练方法， 其特 征在于， 所述方法包括： 获取属于目标风格的标准图像， 所述目标风格为任一 风格； 调用正在训练的图像生成模型， 对原始图像进行风格迁移， 得到目标图像， 所述图像生 成模型用于生成属于所述目标风格的图像； 基于所述目标图像， 确定与所述目标图像的风格匹配的第一样本 图像， 基于所述图像 生成模型当前 的训练次数， 确定所述第一样本图像对应的样本标签， 所述样本标签用于表 示所述第一样本图像的风格与所述目标风格之 间的实际相似度， 所述样本标签与所述图像 生成模型的训练次数正相关； 调用待训练的图像检测模型， 将所述第一样本 图像与所述标准图像进行对比， 得到第 一检测标签， 所述第一检测标签用于表示所述第一样本图像的风格与所述目标风格之 间的 预测相似度； 基于所述样本标签和所述第一检测标签， 训练所述待训练的图像检测模型， 得到训练 完成的图像检测模型， 所述训练完成的图像检测模型用于检测任两个图像的风格之 间的相 似度。 2.根据权利要求1所述的图像检测模型的训练方法， 其特征在于， 所述基于所述目标图 像， 确定与所述目标图像的风格匹配的第一样本图像， 包括： 将所述目标图像确定为所述第一样本图像。 3.根据权利要求2所述的图像 检测模型的训练方法， 其特 征在于， 所述方法还 包括： 确定多个候选图像生成模型， 每个候选图像生成模型具有各自的训练次数和各自的准 确度， 且不同的候选图像生成模型当前 的训练次数不同， 所述准确度表示所述候选图像生 成模型生成的属于所述目标风格的图像的准确程度； 在所述多个候选图像生成模型中， 选取所述准确度小于目标阈值的所述图像生成模 型。 4.根据权利要求1所述的图像检测模型的训练方法， 其特征在于， 所述调用正在训练 的 图像生成模型， 对原 始图像进行风格迁移， 得到目标图像， 包括： 确定多个图像生成模型， 且不同的图像生成模型当前的训练次数不同； 调用所述多个图像生成模型， 分别对所述原 始图像进行风格迁移， 得到多个目标图像； 所述基于所述目标图像， 确定与所述目标图像的风格匹配的第一样本图像， 包括： 将所述多个目标图像进行融合所 得到的图像， 确定为所述第一样本图像； 所述基于所述图像生成模型当前的训练次数， 确定所述第一样本图像对应的样本标 签， 包括： 将所述多个图像生成模型的训练次数进行加权平均， 得到平均训练次数； 对所述平均训练次数进行归一 化处理， 得到所述样本标签。 5.一种图像 检测方法， 其特 征在于， 所述方法包括： 获取属于目标风格的标准图像， 所述目标风格为任一 风格； 将第一图像的风格迁移为所述目标风格， 得到属于所述目标风格的第二图像； 调用训练完成的图像检测模型， 将所述第二图像与所述标准图像进行对比， 得到目标 检测标签， 所述目标检测标签用于表示所述第二图像的风格与所述标准图像的风格之 间的 相似度；权　利　要　求　书 1/2 页 2 CN 114998196 A 2其中， 所述训练完成的图像检测模型是采用如权利要求1至权利要求4任一项所述的图 像检测模型的训练方法所训练的。 6.一种图像 检测模型的训练装置， 其特 征在于， 所述装置包括： 第一获取单元， 被配置为执行获取属于目标风格的标准图像， 所述目标风格为任一风 格； 第二获取单元， 被配置为执行调用正在训练的图像生成模型， 对原始图像进行风格迁 移， 得到目标图像， 所述图像生成模型用于生成属于所述目标风格的图像； 第一确定单元， 被配置为执行基于所述目标图像， 确定与所述目标图像的风格匹配的 第一样本图像， 基于所述图像生成模型当前 的训练次数， 确定所述第一样本图像对应的样 本标签， 所述样本标签用于表示所述第一样本图像的风格与所述目标风格之间的实际相似 度， 所述样本标签与所述图像生成模型的训练次数正相关； 模型训练单元， 被配置为执行调用待训练的图像检测模型， 将所述第一样本 图像与所 述标准图像进行对比， 得到第一检测标签， 所述第一检测标签用于表示所述第一样本图像 的风格与所述目标风格之间的预测相似度； 所述模型训练单元， 还被配置为执行基于所述样本标签和所述第一检测标签， 训练所 述待训练的图像检测模型， 得到训练完成的图像检测模型， 所述训练完成的图像检测模型 用于检测任两个图像的风格之间的相似度。 7.一种图像 检测装置， 其特 征在于， 所述装置包括： 获取单元， 被配置为执 行获取属于目标风格的标准图像， 所述目标风格为任一 风格； 风格迁移单元， 被配置为执行将第一图像的风格迁移为所述目标风格， 得到属于所述 目标风格的第二图像； 图像检测单元， 被配置为执行调用训练完成的图像检测模型， 将所述第二图像与所述 标准图像进行对比， 得到目标检测标签， 所述 目标检测标签用于表示所述第二图像的风格 与所述标准图像的风格之间的相似度； 其中， 所述训练完成的图像检测模型是采用如权利要求1至权利要求4任一项所述的图 像检测模型的训练方法所训练的。 8.一种电子设备， 其特 征在于， 所述电子设备包括： 处理器； 用于存储所述处 理器可执行指令的存 储器； 其中， 所述处理器被配置为执行所述指令， 以实现如权利要求1至权利要求4任一项所 述的图像 检测模型的训练方法， 或者实现如权利要求5所述的图像 检测方法。 9.一种计算机可读存储介质， 当所述计算机可读存储介质中的指令由电子设备的处理 器执行时， 使得所述电子 设备能够执行如权利要求1至权利要求4任一项 所述的图像检测模 型的训练方法， 或者实现如权利要求5所述的图像 检测方法。 10.一种计算机程序产品， 包括计算机程序， 其特征在于， 所述计算机程序被处理器执 行时实现权利要求 1至权利要求4任一项 所述的图像检测模型的训练方法， 或者实现如权利 要求5所述的图像 检测方法。权　利　要　求　书 2/2 页 3 CN 114998196 A 3