[发明专利]视频分类方法、电子设备和存储介质

说明书

本发明实施例提供了一种视频分类方法及装置，该方法包括根据目标视频的组成元素确定各组成元素对应的特征信息；将各组成元素对应的特征信息输入视频分类模型中，得到由视频分类模型输出的分类信息。本方法通过视频分类模型实现对视频的分类，该视频分类模型包括用于对特征信息进行升维处理的升维层和根据作为训练参数的聚类中心矩阵和升维后的特征信息进行运算的聚类运算层。通过对特征信息进行升维处理和聚类运算层的运算过程，能够基于聚类中心矩阵中每一列向量所表示的聚类中心对特征信息进行聚类式的分析，提取有利于确定视频所属类别的特征，提高视频分类的准确性，提高了对视频分类的分类效率。

技术领域

本发明涉及机器学习和视频分析技术领域，尤其是涉及一种视频分类方法、电子设备和存储介质。

背景技术

视频分类有助于进行视频的检索和管理，通常通过对视频添加的标签表示视频所属的类别。传统的方法多通过人工标注的方式对视频进行分类。然而，随着互联网技术的发展，越来越多的视频，尤其是短视频，出现在网络，例如，个人用户上传的短视频。这些短视频涉及动漫、影视、饮食、文娱、体育、游戏等多个类别。如果通过人工标注的方式进行分类，不仅需要消耗大量人力成本，且容易被受个人主观因素影响导致分类不全面和不准确。

可见，通过人工标注的方式对视频进行分类不仅效率低，且容易导致分类不准确。

发明内容

针对现有技术存在的问题，本发明实施例提供一种视频分类方法、装置、电子设备及存储介质。

第一方面，本发明实施例提供一种视频分类方法，包括：

根据目标视频的组成元素确定各组成元素对应的特征信息，所述组成元素包括图像，声音，和/或字幕；

将各组成元素对应的特征信息输入视频分类模型中，得到由所述视频分类模型输出的分类信息，所述分类信息用于表征所述目标视频的类别；

其中，所述视频分类模型为将根据样本视频确定的特征信息和所述样本视频的类别标签作为输入，通过机器学习训练得到的，用于对视频进行分类的模型；

所述视频分类模型包括升维层和聚类运算层，所述升维层用于对所述特征信息进行升维处理，所述聚类运算层用于根据作为训练参数的聚类中心矩阵和升维后的所述特征信息进行运算。

进一步地，所述将各组成元素对应的特征信息输入视频分类模型中，得到由所述视频分类模型输出的分类信息，包括：

将各组成元素的所述特征信息输入到所述升维层，由所述升维层输出升维后的特征信息；

将升维后的所述特征信息输入到所述聚类运算层，由所述聚类运算层输出第一矩阵；

将所述第一矩阵输入到所述视频分类模型的权重聚合层，由所述权重聚合层输出第二矩阵；

将所述第二矩阵输入到所述视频分类模型的分类层，由所述分类层输出所述分类信息；

其中，所述权重聚合层包括至少一个卷积子层和至少一个激活函数，所述分类信息包括所述目标视频的多个类别标签。

进一步地，所述聚类运算层包括多个聚类子层，所述聚类子层的个数与视频的组成元素的个数相同，相应地，所述将升维后的所述特征信息输入到所述聚类运算层，由所述聚类运算层输出第一矩阵，包括：

将升维后的各组成元素对应的特征信息分别输入到对应的聚类子层，由各聚类子层输出各组成元素对应的聚类运算矩阵；

将各组成元素对应的聚类运算矩阵进行拼接，得到所述第一矩阵；

其中，所述聚类子层用于根据所述特征信息和属于所述聚类子层的聚类中心矩阵和初始权重矩阵进行运算。