[发明专利]视频分类模型的生成方法、装置、设备和存储介质

说明书

本公开公开了一种视频分类模型的生成方法、装置、设备和存储介质，涉及计算机技术领域，具体涉及计算机视觉、深度学习等人工智能技术领域。视频分类模型的生成方法包括：获取图像分类模型，所述图像分类模型根据图像分类数据集，采用第一知识蒸馏网络生成；根据视频分类数据集，采用第二知识蒸馏网络，对所述图像分类模型进行训练，以生成视频分类模型。本公开可以提高视频分类模型的分类效果。

技术领域

本公开涉及计算机技术领域，具体涉及计算机视觉、深度学习等人工智能技术领域，尤其涉及一种视频分类模型的生成方法、装置、设备和存储介质。

背景技术

人工智能(Artificial Intelligence，AI)是研究使计算机来模拟人的某些思维过程和智能行为(如学习、推理、思考、规划等)的学科，既有硬件层面的技术也有软件层面的技术。人工智能硬件技术一般包括如传感器、专用人工智能芯片、云计算、分布式存储、大数据处理等技术；人工智能软件技术主要包括计算机视觉技术、语音识别技术、自然语言处理技术以及机器学习/深度学习、大数据处理技术、知识图谱技术等几大方向。

随着移动设备的普及和硬件性能的大幅提升，用视频记录生活已经逐渐成为人们的日常行为习惯。随着视频数据的极速增长，可以采用视频分类模型对视频进行分类，以便基于分类对视频进行存储、管理等。为了降低参数量，提高实用性，可以采用知识蒸馏的方式生成视频分类模型。知识蒸馏网络包括知识迁移模型和知识学习模型。

相关技术中，仅进行一次基于视频分类数据集的知识蒸馏过程，将待生成的视频分类模型作为知识学习模型，采用知识蒸馏的方式，将知识迁移模型的输出作为监督信息，训练知识学习模型后生成视频分类模型。

发明内容

本公开提供了一种视频分类模型的生成方法、装置、设备和存储介质。

根据本公开的一方面，提供了一种视频分类模型的生成方法，包括：获取图像分类模型，所述图像分类模型根据图像分类数据集，采用第一知识蒸馏网络生成；根据视频分类数据集，采用第二知识蒸馏网络，对所述图像分类模型进行训练，以生成视频分类模型。

根据本公开的另一方面，提供了一种视频分类模型的生成装置，包括：获取模块，用于获取图像分类模型，所述图像分类模型根据图像分类数据集，采用第一知识蒸馏网络生成；生成模块，用于根据视频分类数据集，采用第二知识蒸馏网络，对所述图像分类模型进行训练，以生成视频分类模型。

根据本公开的另一方面，提供了一种电子设备，包括：至少一个处理器；以及与所述至少一个处理器通信连接的存储器；其中，所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行如上述任一方面的任一项所述的方法。

根据本公开的另一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行根据上述任一方面的任一项所述的方法。

根据本公开的另一方面，提供了一种计算机程序产品，包括计算机程序，所述计算机程序在被处理器执行时实现根据上述任一方面的任一项所述的方法。

根据本公开的技术方案，可以提高视频分类模型的分类效果。

应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。

附图说明

附图用于更好地理解本方案，不构成对本公开的限定。其中：

图1是根据本公开第一实施例的示意图；

图2是根据本公开第二实施例的示意图；

图3是根据本公开第三实施例的示意图；

图4是根据本公开第四实施例的示意图；