[发明专利]影视角色分类方法、装置、计算机设备和存储介质

说明书

本申请涉及一种影视角色分类方法、装置、计算机设备和存储介质。方法包括：获取来源于至少一个影视视频的目标图像；对各目标图像分别进行主体检测，并基于主体检测结果从目标图像中裁剪出角色图像；对各角色图像分别进行特征提取，得到对应的图像特征；将图像特征按照对应角色图像所属的影视视频，划分成至少一个特征子集；每个特征子集中的图像特征所对应的角色图像，来源于同一个影视视频；对于每个特征子集，对相应特征子集中的图像特征进行聚类得到至少一个聚类簇，并将同一聚类簇下图像特征所对应的角色图像划分至同一角色类别。采用本方法能够基于人工智能技术实现主体检测、特征提取和聚类分析，从而提高影视角色分类的效率。

技术领域

本申请涉及计算机技术领域，特别是涉及一种影视角色分类方法、装置、计算机设备和存储介质。

背景技术

随着社会和经济的发展，各种影视剧越来越多，对影视剧中出现的各个对象进行角色的分类这一需求也越来越大。对影视剧中出现的对象进行角色的分类，可以方便用户查找某一对象的角色，或者获取某一角色的剧照等等。

传统的影视角色分类方法，通常是由工作人员手动对影视剧中出现的每一个对象进行角色的分类，存在角色分类效率低的问题。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高效率的影视角色分类方法、装置、计算机设备和存储介质。

一种影视角色分类方法，所述方法包括：

获取来源于至少一个影视视频的目标图像；

对各所述目标图像分别进行主体检测，并基于主体检测结果从所述目标图像中裁剪出角色图像；

对各所述角色图像分别进行特征提取，得到对应的图像特征；

将所述图像特征按照对应角色图像所属的影视视频，划分成至少一个特征子集；每个特征子集中的图像特征所对应的角色图像，来源于同一个影视视频；

对于每个特征子集，对相应特征子集中的图像特征进行聚类得到至少一个聚类簇，并将同一聚类簇下图像特征所对应的角色图像划分至同一角色类别。

一种影视角色分类装置，所述装置包括：

获取模块，用于获取来源于至少一个影视视频的目标图像；

检测模块，用于对各所述目标图像分别进行主体检测，并基于主体检测结果从所述目标图像中裁剪出角色图像；

特征提取模块，用于对各所述角色图像分别进行特征提取，得到对应的图像特征；

分类模块，用于将所述图像特征按照对应角色图像所属的影视视频，划分成至少一个特征子集；每个特征子集中的图像特征所对应的角色图像，来源于同一个影视视频；对于每个特征子集，对相应特征子集中的图像特征进行聚类得到至少一个聚类簇，并将同一聚类簇下图像特征所对应的角色图像划分至同一角色类别。

在其中一个实施例中，所述获取模块还用于获取至少一个影视视频；对于每个影视视频，在所述影视视频的播放时间轴上，每间隔预设播放时长抽取出至少一个目标图像。

在其中一个实施例中，所述检测模块还用于对各所述目标图像分别进行主体检测，得到主体检测结果；若所述主体检测结果表示所述目标图像中存在角色对象，则基于所述主体检测结果确定所述角色对象所在的角色区域；基于所述角色区域，从所述目标图像中裁剪出包括有角色对象的角色图像。

在其中一个实施例中，所述装置还包括调整模块，用于获取主体检测模型的第一输入图像尺寸；若所述目标图像的尺寸与所述第一输入图像尺寸不匹配，则将所述目标图像的尺寸调整至所述第一输入图像尺寸，得到第一图像；将所述第一图像作为新的目标图像；所述检测模块还用于对各所述目标图像分别进行主体检测。