[发明专利]用于生成图像类别检测模型的方法和装置在审
申请号: | 201811075020.X | 申请日: | 2018-09-14 |
公开(公告)号: | CN109191453A | 公开(公告)日: | 2019-01-11 |
发明(设计)人: | 肖梅峰;徐珍琦;王长虎 | 申请(专利权)人: | 北京字节跳动网络技术有限公司 |
主分类号: | G06T7/00 | 分类号: | G06T7/00;G06N3/04 |
代理公司: | 北京英赛嘉华知识产权代理有限责任公司 11204 | 代理人: | 王达佐;马晓亚 |
地址: | 100041 北京市石景山区*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本申请实施例公开了用于生成图像类别检测模型的方法和装置。该方法的一具体实施方式包括:获取样本集;从该样本集中提取样本,执行如下训练步骤:将所提取的样本中的样本图像分别输入至预先训练的第一图像类别检测模型和初始模型;基于该第一图像类别检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所提取的样本的损失值;基于该损失值确定初始模型是否训练完成;响应于确定初始模型训练完成,将训练后的初始模型确定为第二图像类别检测模型。该实施方式能够得到一种可以适用于移动端的图像类别检测模型,且该方法丰富了模型的生成方式。 | ||
搜索关键词: | 初始模型 图像类别 样本 检测 方法和装置 类别检测 生成图像 目标层 输出 标注信息 生成方式 训练步骤 样本图像 预测结果 样本集 响应 移动 申请 | ||
【主权项】:
1.一种用于生成图像类别检测模型的方法,包括:获取样本集,其中,所述样本集中的样本包括样本图像和用于表征样本图像的类别的标注信息;从所述样本集中提取样本,执行如下训练步骤:将所提取的样本中的样本图像分别输入至初始模型和预先训练的第一图像类别检测模型;基于所述第一图像类别检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所提取的样本的损失值;基于所述损失值与目标值的比较,确定初始模型是否训练完成;响应于确定初始模型训练完成,将训练后的初始模型确定为第二图像类别检测模型。
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京字节跳动网络技术有限公司,未经北京字节跳动网络技术有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201811075020.X/,转载请声明来源钻瓜专利网。