standard download
(19)国家知识产权局 (12)发明 专利 (10)授权公告 号 (45)授权公告日 (21)申请 号 202210436659.6 (22)申请日 2022.04.25 (65)同一申请的已公布的文献号 申请公布号 CN 114549938 A (43)申请公布日 2022.05.27 (73)专利权人 广州市玄武无线科技股份有限公 司 地址 510620 广东省广州市天河区体 育西 路109号23Bd房 (72)发明人 丁明 王杰 钟忞盛  (74)专利代理 机构 广州三环 专利商标代理有限 公司 44202 专利代理师 李家平 (51)Int.Cl. G06V 10/774(2022.01)G06V 20/80(2022.01) G06F 16/51(2019.01) G06F 16/55(2019.01) G06F 16/583(2019.01) G06K 9/62(2022.01) G06V 10/74(2022.01) (56)对比文件 CN 114155388 A,2022.03.08 CN 112949780 A,2021.0 6.11 WO 2021087985 A1,2021.0 5.14 贺周雨 等.基 于SHN模型的商品图像 检索方 法. 《计算机 工程与科 学》 .2019,第41卷(第1 1 期), 审查员 曹春晓 (54)发明名称 模型训练方法、 图像信息管理方法、 图像识 别方法及装置 (57)摘要 本发明公开了一种模 型训练方法、 图像信息 管理方法、 图像识别方法及装置。 所述模型训练 方法包括: 对初始图像处理模型进行随机初始 化; 所述初始图像处理模型包括特征提取分支和 原型构建分支; 固定原型构建分支参数, 将训练 图像集输入至初始图像处理模型, 得到第一输出 结果; 根据第一输出结果确定损失函数, 利用损 失函数对特征提取分支的初始化参数进行更新; 固定特征提取分支更新后的参数, 将训练图像集 输入至初始图像处理模型中, 得到第二输出结 果; 根据第二输出结果确定损失函数, 利用损失 函数对原型构建分支的初始化参数进行更新; 根 据特征提取分支更新后的参数构建商品指纹提 取模型, 以使商品指纹提取模型输出的商品指纹 具备更好的区分性。 权利要求书3页 说明书6页 附图2页 CN 114549938 B 2022.09.09 CN 114549938 B 1.一种模型训练方法, 其特 征在于, 包括: 对初始图像处理模型的参数进行随机初始化; 其中, 所述初始图像处理模型包括特征 提取分支和原型构建 分支; 固定所述原型构建分支 的初始化参数, 将训练图像集输入至所述初始图像处理模型, 得到第一输出 结果; 根据所述第 一输出结果确定损失函数的值, 利用所述损失函数的值对所述特征提取分 支的初始化 参数进行 更新; 所述损失函数包括类别损失和类内类间损失; 其中, 所述类别损失定义为 ; 式中,fx表示所述 特征提取分支输出的商品指纹, e表示所述原型构建分支输出的指纹原型, <  , >表示内积, i表示商品类别, T表示温度系数, N表示图像样本数量; 所述类内类间损失定义为 ; 式 中, 表示边界阈值 , zp表示第一输出结果或第二输出结果中与所述特征提取分支输 出的商品指纹类别相同的结果, zn表示第一输出结果或第二输出结果中与所述特征提取分 支输出的商品指纹类别不相同的结果; 固定所述特征提取分支更新后的参数, 将所述训练图像集输入至特征提取分支参数更 新后的初始图像处 理模型中, 得到第二输出 结果; 根据所述第 二输出结果确定所述损失函数的值, 利用所述损失函数的值对所述原型构 建分支的初始化 参数进行 更新; 根据所述特 征提取分支更新后的参数构建商品指纹提取模型。 2.根据权利要求1所述的模型训练方法, 其特征在于, 所述第 一输出结果和所述第 二输 出结果均包括: 所述特征提取分支输出的商品指纹和所述原型构建 分支输出的指纹原型。 3.一种图像信息管理方法, 其特 征在于, 包括: 获取场景图像集, 利用图像 检测模型获取 所述场景图像集中的全量商品图像; 根据商品指纹提取模型提取 所述商品图像中的商品指纹, 得到商品指纹集; 其中, 所述商品指纹提取模型为采用如权利要求1~2任一项所述的模型训练方法训练出的模 型; 确定所述商 品指纹集的商 品类别, 利用聚类算法对各所述商 品类别对应的商品指纹进 行压缩; 将压缩后的商品指纹存 储至商品指纹数据库。 4.根据权利要求3所述的图像信 息管理方法, 其特征在于, 所述利用聚类算法对各所述 商品类别对应的商品指纹进行压缩, 具体为: 确定所述商品指纹集中各商品指纹的商品类别; 根据预设的数量参数N, 利用聚类算法依次将各商品类别下的商品指纹压缩为N个类 簇, 并将N个所述类簇的类簇中心设置为对应商品类别的目标商品指纹; 其中, N为 非零自然 数;权 利 要 求 书 1/3 页 2 CN 114549938 B 2将各所述商品类别的目标商品指纹存 储至商品指纹数据库。 5.一种图像识别方法, 其特 征在于, 包括: 获取目标商品图像, 利用商品指纹提取模型提取 所述目标商品图像的目标商品指纹; 基于所述目标商 品指纹遍历商 品指纹数据库中的基准指纹, 将与所述目标商品指纹相 似度最高的基准指纹所对应的商品类别确定为识别结果; 其中, 所述商品指纹提取模型为采用如权利要求1~2任一项所述的模型训练方法训练出的模 型; 所述商品指纹数据库由如权利要求3~4任一项所述的图像信息管理方法得到 。 6.一种模型训练装置, 其特 征在于, 包括: 初始化模块, 用于对初始图像处理模型的参数进行随机初始化; 其中, 所述初始图像处 理模型包括特 征提取分支和原型构建 分支; 第一训练模块, 用于固定所述原型构建分支 的初始化参数, 将训练图像集输入至所述 初始图像处 理模型, 得到第一输出 结果; 根据所述第 一输出结果确定损失函数的值, 利用所述损失函数的值对所述特征提取分 支的初始化 参数进行 更新; 所述损失函数包括类别损失和类内类间损失; 其中, 所述类别损失定义为 ; 式中,fx表示所述 特征提取分支输出的商品指纹, e表示所述原型构建分支输出的指纹原型, <  , >表示内积, i表示商品类别, T表示温度系数, N表示图像样本数量; 所述类内类间损失定义为 ; 式 中, 表示边界阈值 , zp表示第一输出结果或第二输出结果中与所述特征提取分支输 出的商品指纹类别相同的结果, zn表示第一输出结果或第二输出结果中与所述特征提取分 支输出的商品指纹类别不相同的结果; 第二训练模块, 用于固定所述特征提取分支更新后的参数, 将所述训练图像集输入至 特征提取分支参数 更新后的初始图像处 理模型中, 得到第二输出 结果; 根据所述第 二输出结果确定所述损失函数的值, 利用所述损失函数的值对所述原型构 建分支的初始化 参数进行 更新; 确定模块, 用于根据所述特 征提取分支更新后的参数构建商品指纹提取模型。 7.根据权利要求6所述的模型训练装置, 其特征在于, 所述第 一输出结果和所述第 二输 出结果均包括: 所述特征提取分支输出的商品指纹和所述原型构建 分支输出的指纹原型。 8.一种数据处 理设备, 其特 征在于, 包括: 处理器, 所述处理器和存储器耦合, 所述存储器存储有程序, 所述程序由所述处理器执 行, 使得所述数据处理设备执行如权利要求 1~2任一项所述的模型训练方法, 或如权利要求 3~4任一项所述的图像信息管理方法, 或如权利要求5所述的图像识别方法。 9.一种计算机存储介质, 其特征在于, 所述计算机存储介质存储有计算机指令, 所述计 算机指令用于执行如权利要求1~2任一项所述的模型训练方法, 或如权利要求3~4任一项所权 利 要 求 书 2/3 页 3 CN 114549938 B 3

PDF文档 专利 模型训练方法、图像信息管理方法、图像识别方法及装置

文档预览
中文文档 12 页 50 下载 1000 浏览 0 评论 0 收藏 3.0分
温馨提示:本文档共12页,可预览 3 页,如浏览全部内容或当前文档出现乱码,可开通会员下载原始文档
专利 模型训练方法、图像信息管理方法、图像识别方法及装置 第 1 页 专利 模型训练方法、图像信息管理方法、图像识别方法及装置 第 2 页 专利 模型训练方法、图像信息管理方法、图像识别方法及装置 第 3 页
下载文档到电脑,方便使用
本文档由 SC 于 2024-02-24 01:00:08上传分享
站内资源均来自网友分享或网络收集整理,若无意中侵犯到您的权利,敬请联系我们微信(点击查看客服),我们将及时删除相关资源。