[发明专利]一种标签提取方法及装置

说明书

本申请涉及计算机技术领域，并涉及人工智能的计算机视觉、机器学习等技术，尤其涉及一种标签提取方法及装置，获取待提取视频和所述待提取视频的视频标题；提取所述视频标题的标题特征信息；提取所述待提取视频中的各个目标视频帧的图像特征信息，并分别对所述各个目标视频帧，根据目标视频帧的图像特征信息和所述标题特征信息进行目标检测，获得所述各个目标视频帧的标签；根据所述各个目标视频帧的标签，确定所述待提取视频的标签，这样，融合标题特征信息和图像特征信息进行标签提取，可以提高视频标签提取的准确性。

技术领域

本申请涉及计算机技术领域，尤其涉及一种标签提取方法及装置。

背景技术

视频标签提取是指识别视频中描述的主要物体和内容，视频标签可应用于视频的分类、推荐及检索等任务中，相关技术中视频标签提取方法通常仅是依据单一信息来提取视频标签，但是单独使用一种信息受到影响较大，准确性较低，容易导致提取的标签错误。

发明内容

本申请实施例提供一种标签提取方法及装置，以提高视频标签提取的准确性。

本申请实施例提供的具体技术方案如下：

本申请一个实施例提供了一种标签提取方法，包括：

获取待提取视频和所述待提取视频的视频标题；

提取所述视频标题的标题特征信息；

提取所述待提取视频中的各个目标视频帧的图像特征信息，并分别对所述各个目标视频帧，根据目标视频帧的图像特征信息和所述标题特征信息进行目标检测，获得所述各个目标视频帧的标签；

根据所述各个目标视频帧的标签，确定所述待提取视频的标签。

本申请另一个实施例提供了一种标签提取装置，包括：

获取模块，用于获取待提取视频和所述待提取视频的视频标题；

提取模块，用于提取所述视频标题的标题特征信息；

处理模块，用于提取所述待提取视频中的各个目标视频帧的图像特征信息，并分别对所述各个目标视频帧，根据目标视频帧的图像特征信息和所述标题特征信息进行目标检测，获得所述各个目标视频帧的标签；

确定模块，用于根据所述各个目标视频帧的标签，确定所述待提取视频的标签。

可选的，提取所述视频标题的标题特征信息时，提取模块具体用于：

基于已训练的文本编码模型，以所述视频标题为输入，依次通过所述文本编码模型的卷积层和池化层，对所述视频标题进行卷积和压缩操作，获得所述视频标题的标题特征信息。

可选的，提取所述待提取视频中的各个目标视频帧的图像特征信息，并分别对所述各个目标视频帧，根据目标视频帧的图像特征信息和所述标题特征信息进行目标检测，获得所述各个目标视频帧的标签时，处理模块具体用于：

基于已训练的目标检测模型，提取所述待提取视频中的各个目标视频帧的图像特征信息，并分别对所述各个目标视频帧，根据目标视频帧的图像特征信息和所述标题特征信息进行目标检测，获得所述各个目标视频帧的标签，所述目标检测模型包括图像特征提取模块和目标检测模块，其中：

所述图像特征提取模块分别以所述各个目标视频帧为输入，对所述各个目标视频帧进行卷积操作，获得所述各个目标视频帧的图像特征信息；

所述目标检测模块以所述标题特征信息和所述各个目标视频帧的图像特征信息为输入，分别对所述各个目标视频帧进行目标检测，获得所述各个目标视频帧中检测到的目标所属的类别和类别置信度；并分别将所述各个目标视频帧中检测到的目标所属的类别，作为对应目标视频帧的标签。