[发明专利]用于生成模型的方法和装置有效
申请号: | 201810617804.4 | 申请日: | 2018-06-15 |
公开(公告)号: | CN108805091B | 公开(公告)日: | 2021-08-10 |
发明(设计)人: | 李伟健;王长虎 | 申请(专利权)人: | 北京字节跳动网络技术有限公司 |
主分类号: | G06K9/00 | 分类号: | G06K9/00;G06K9/62;G06N3/04;G06N3/08 |
代理公司: | 北京英赛嘉华知识产权代理有限责任公司 11204 | 代理人: | 王达佐;马晓亚 |
地址: | 100041 北京市石景山区*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 用于 生成 模型 方法 装置 | ||
1.一种用于生成模型的方法,包括:
获取训练样本集合,以及将训练样本集合划分成预设数量个训练样本组,其中,训练样本包括样本视频和针对样本视频预先标注的样本识别结果,所述样本视频为对样本对象进行拍摄所获得的视频,所述样本识别结果用于指示所述样本视频是否为对显示所述样本对象的屏幕进行拍摄所获得的视频;
对于所述预设数量个训练样本组中的训练样本组,将该组训练样本中的训练样本的样本视频作为输入,将所输入的样本视频所对应的样本识别结果作为期望输出,利用机器学习方法训练得到该组训练样本所对应的初始视频识别模型,具体包括:基于所述预设数量个训练样本组的优劣程度,从所述预设数量个训练样本组中选取最优的训练样本组作为候选训练样本组,以及基于候选训练样本组和初始模型,执行以下训练步骤:将候选训练样本组中的训练样本的样本视频作为输入,将所输入的样本视频所对应的样本识别结果作为期望输出,利用机器学习方法对初始模型进行训练,获得初始视频识别模型;确定所述预设数量个训练样本组中是否存在未被选取的训练样本组;响应于确定存在未被选取的训练样本组,从所述未被选取的训练样本组中选取最优的训练样本组作为新的候选训练样本组,将最近一次获得的初始视频识别模型作为新的初始模型,继续执行所述训练步骤;
基于所得到的初始视频识别模型,生成视频识别模型。
2.根据权利要求1所述的方法,其中,所述对于所述预设数量个训练样本组中的训练样本组,将该组训练样本中的训练样本的样本视频作为输入,将所输入的样本视频所对应的样本识别结果作为期望输出,利用机器学习方法训练得到该组训练样本所对应的初始视频识别模型,包括:
响应于确定不存在未被选取的训练样本组,获得预设数量个初始视频识别模型。
3.根据权利要求1所述的方法,其中,所述对于所述预设数量个训练样本组中的训练样本组,将该组训练样本中的训练样本的样本视频作为输入,将所输入的样本视频所对应的样本识别结果作为期望输出,利用机器学习方法训练得到该组训练样本所对应的初始视频识别模型,包括:
确定用于表征所述预设数量个训练样本组的优劣程度的数值;
基于所确定的数值,从所述预设数量个训练样本组中选取最优的训练样本组作为候选训练样本组,以及基于候选训练样本组和初始模型,执行以下训练步骤:将候选训练样本组中的训练样本的样本视频作为输入,将所输入的样本视频所对应的样本识别结果作为期望输出,利用机器学习方法对初始模型进行训练,获得初始视频识别模型;确定所述预设数量个训练样本组中是否存在未被选取的训练样本组;响应于确定不存在未被选取的训练样本组,获得预设数量个初始视频识别模型;
响应于确定存在未被选取的训练样本组,基于所确定的数值,从未被选取的训练样本组中选取最优的训练样本组作为新的候选训练样本组,将最近一次获得的初始视频识别模型作为新的初始模型,继续执行所述训练步骤。
4.根据权利要求3所述的方法,其中,所述确定用于表征所述预设数量个训练样本组的优劣程度的数值,包括:
获取预先设置的验证样本集合,其中,验证样本包括验证用视频和针对验证用视频预先标注的验证用识别结果;
对于所述预设数量个训练样本组中的训练样本组,执行以下步骤:将该组训练样本中的训练样本的样本视频作为输入,将所输入的样本视频所对应的样本识别结果作为输出,利用机器学习方法训练得到该组训练样本所对应的待验证视频识别模型;将验证样本集合中的验证样本的验证用视频输入该组训练样本所对应的待验证视频识别模型,获得实际识别结果,确定实际识别结果相对于所输入的验证用视频所对应的验证用识别结果的损失值,基于所确定的损失值,生成用于表征该组训练样本组的优劣程度的数值。
5.根据权利要求3-4之一所述的方法,其中,所述基于所得到的初始视频识别模型,生成视频识别模型,包括:
基于所确定的数值,为所获得的初始视频识别模型分配权重;
基于所分配的权重,对所获得的初始视频识别模型进行融合,生成视频识别模型。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京字节跳动网络技术有限公司,未经北京字节跳动网络技术有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201810617804.4/1.html,转载请声明来源钻瓜专利网。