[发明专利]模型训练、视频处理方法,装置,设备以及存储介质有效
| 申请号: | 202110926860.8 | 申请日: | 2021-08-12 |
| 公开(公告)号: | CN113627354B | 公开(公告)日: | 2023-08-08 |
| 发明(设计)人: | 吴文灏;黄登 | 申请(专利权)人: | 北京百度网讯科技有限公司 |
| 主分类号: | G06V20/40 | 分类号: | G06V20/40;G06V10/74;G06V10/774;G06V10/82 |
| 代理公司: | 北京英赛嘉华知识产权代理有限责任公司 11204 | 代理人: | 王达佐;马晓亚 |
| 地址: | 100085 北京市*** | 国省代码: | 北京;11 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 模型 训练 视频 处理 方法 装置 设备 以及 存储 介质 | ||
1.一种模型训练方法,包括:
从样本视频集合中选取出第一样本视频和第二样本视频,其中,所述第一样本视频和所述第二样本视频的外观相似度大于第二预设阈值;
从所述第一样本视频中选取连续的多个视频帧;
将所述多个视频帧划分为数量相同的两个视频片段,得到第一视频片段和第二视频片段;
确定所述第二视频片段的每秒显示帧数;
以所述每秒显示帧数对所述第二样本视频进行采样,得到第三视频片段;
利用目标模型分别提取所述第一视频片段、所述第二视频片段以及所述第三视频片段的特征,得到所述第一视频片段的第一特征、所述第二视频片段的第二特征以及所述第三视频片段的第三特征;
根据所述第一特征与所述第二特征之间的第一距离、所述第二特征和所述第三特征之间的第二距离,确定损失函数;
根据所述损失函数,训练所述目标模型。
2.根据权利要求1所述的方法,其中,所述利用目标模型分别提取所述第一视频片段、所述第二视频片段以及所述第三视频片段的特征,得到所述第一视频片段的第一特征、所述第二视频片段的第二特征以及所述第三视频片段的第三特征,包括:
对所述第一视频片段和所述第二视频片段进行数据增强;
利用所述目标模型对数据增强后的第一视频片段、数据增强后的第二视频片段和所述第三视频片段进行特征提取,得到所述第一特征、所述第二特征和第三特征。
3.根据权利要求1所述的方法,其中,所述方法还包括:
根据下游任务的样本数据,对训练好的目标模型进行微调。
4.一种视频处理方法,包括:
获取目标视频;
利用通过权利要求1-3任一项所述的模型训练方法训练得到的目标模型提取所述目标视频的特征,确定所述目标视频的目标特征;
根据所述目标特征,对所述目标视频进行处理。
5.一种模型训练装置,包括:
样本视频选取单元,被配置成从样本视频集合中选取出第一样本视频和第二样本视频,其中,所述第一样本视频和所述第二样本视频的外观相似度大于第二预设阈值;
视频帧选取单元,被配置成从所述第一样本视频中选取连续的多个视频帧;
视频帧划分单元,被配置成将所述多个视频帧划分为数量相同的两个视频片段,得到第一视频片段和第二视频片段;
显示帧数确定单元,被配置成确定所述第二视频片段的每秒显示帧数;
视频帧采样单元,被配置成以所述每秒显示帧数对所述第二样本视频进行采样,得到第三视频片段;
视频特征提取单元,被配置成利用目标模型分别提取所述第一视频片段、所述第二视频片段以及所述第三视频片段的特征,得到所述第一视频片段的第一特征、所述第二视频片段的第二特征以及所述第三视频片段的第三特征;
损失函数确定单元,被配置成根据所述第一特征与所述第二特征之间的第一距离、所述第二特征和所述第三特征之间的第二距离,确定损失函数;
目标模型训练单元,被配置成根据所述损失函数,训练所述目标模型。
6.根据权利要求5所述的装置,其中,所述视频特征提取单元进一步被配置成:
对所述第一视频片段和所述第二视频片段进行数据增强;
利用所述目标模型对数据增强后的第一视频片段、数据增强后的第二视频片段和所述第三视频片段进行特征提取,得到所述第一特征、所述第二特征和第三特征。
7.根据权利要求5所述的装置,其中,所述装置还包括微调单元,被配置成:
根据下游任务的样本数据,对训练好的目标模型进行微调。
8.一种视频处理装置,包括:
视频获取单元,被配置成获取目标视频;
特征提取单元,被配置成利用通过权利要求1-3任一项所述的模型训练方法训练得到的目标模型提取所述目标视频的特征,确定所述目标视频的目标特征;
视频处理单元,被配置成根据所述目标特征,对所述目标视频进行处理。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京百度网讯科技有限公司,未经北京百度网讯科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202110926860.8/1.html,转载请声明来源钻瓜专利网。





