一种用于视频相似度学习模型的训练方法

    公开(公告)号:CN117746302A

    公开(公告)日:2024-03-22

    申请号:CN202410166668.7

    申请日:2024-02-06

    Abstract: 本发明提供了用于视频相似度学习模型的训练方法,包括:获取训练集,包括多个原始样本和指示对应样本类别的标签,原始样本为一段视频且包括多个视频帧;利用训练集多次迭代训练模型,得到经训练的模型,每次包括:针对每个类别,将该类别的一个原始样本作为查询样本,其他原始样本与查询样本构成样本对,得到多个样本对;利用模型提取样本对中样本的特征向量,包括样本的每个视频帧的块级特征序列;基于样本对构建多对视频帧,根据样本的多个视频帧的块级特征序列确定多对视频帧的帧级相似度,根据该帧级相似度确定样本对中两个样本间的视频级相似度;利用预设的损失函数根据查询样本对应的多个样本对的视频级相似度计算的总损失更新模型参数。

Patent Agency Ranking