[发明专利]卡通形象分类方法、装置、电子设备及存储介质

权利要求书

1.一种卡通形象分类方法，其特征在于，所述方法包括：

将包含待分类卡通形象的图像输入特征提取网络模型，得到所述待分类卡通形象的图像特征；

将多个卡通形象关系网络分别输入特征关系网络模型，得到所述多个卡通形象关系网络分别对应的中心节点特征；

基于所述图像特征与所述多个卡通形象关系网络分别对应的中心节点特征，输出所述待分类卡通形象的分类结果。

2.根据权利要求1所述的卡通形象分类方法，其特征在于，所述卡通形象关系网络是通过以下步骤构建的：

以各个卡通形象作为节点，构建所述各个卡通形象的节点之间的多个关系网络；

获取所述各个卡通形象包含的关系信息，并且利用所述特征提取网络模型提取所述各个卡通形象分别对应的图像特征，所述关系信息包括：人物关系与剧情关系；

在所述各个卡通形象分别对应的节点中，输入所述各个卡通形象分别对应的图像特征以及所述各个卡通形象分别包含的关系信息，得到所述多个卡通形象关系网络。

3.据权利要求2所述的卡通形象分类方法，其特征在于，在所述得到所述多个卡通形象关系网络之后，所述方法还包括：

将包含所述中心节点特征的中心节点与目标卡通形象关系网络中的其他节点建立已知连接关系，得到卡通社团网络；

将所述卡通社团网络输入所述特征关系网络模型，得到所述中心节点与所述目标卡通形象关系网络中其他节点的未知连接关系；

利用所述未知连接关系对所述卡通社团网络进行更新。

4.根据权利要求1所述的卡通形象分类方法，其特征在于，所述基于所述图像特征与所述多个卡通形象关系网络分别对应的中心节点特征，输出所述待分类卡通形象的分类结果，包括：

分别计算所述图像特征与各个所述中心节点特征的相似度；

确定与所述图像特征相似度最大的中心节点特征为目标节点特征；

将所述目标节点特征所属的卡通形象关系网络对应的作品确定为所述待分类卡通形象所属的作品；

将所述待分类卡通形象所属的作品进行输出。

5.根据权利要求4所述的卡通形象分类方法，其特征在于，所述分别计算所述图像特征与各个所述中心节点特征的相似度，包括：

将所述图像特征与各个所述中心节点特征分别映射至向量空间，得到所述图像特征的向量与各个所述中心节点特征的向量；

计算所述图像特征的向量与各个所述中心节点特征的向量之间的相似度参数，所述相似度参数包括以下参数的一种或多种：欧氏距离、余弦、点积；

基于所述相似度参数，计算所述图像特征与各个所述中心节点特征的相似度。

6.根据权利要求1所述的卡通形象分类方法，其特征在于，所述特征提取网络模型通过以下步骤训练得到：

获取卡通图像数据集；

剔除所述卡通图像数据集中的非卡通形象，得到卡通形象数据集；

对所述卡通形象数据集进行训练，得到所述特征提取网络模型。

7.根据权利要求1所述的卡通形象分类方法，其特征在于，所述特征关系网络模型是通过以下步骤训练得到的：

将所述卡通形象关系网络包含的网络参数输入三层图卷积神经网络，得到所述卡通形象关系网络中各个节点的预测分类结果；

利用损失函数计算所述各个节点的预测分类结果与所述各个节点的实际分类结果的误差；

基于所述误差对所述三层图卷积神经网络中各个隐含层的权重系数进行更新，得到所述特征关系网络模型。

8.一种卡通形象分类装置，其特征在于，包括：

第一提取模块：用于将包含待分类卡通形象的图像输入特征提取网络模型，得到所述待分类卡通形象的图像特征；

第二提取模块：用于将多个卡通形象关系网络分别输入特征关系网络模型，得到所述多个卡通形象关系网络分别对应的中心节点特征；

分类模块：用于基于所述图像特征与所述多个卡通形象关系网络分别对应的中心节点特征，输出所述待分类卡通形象的分类结果。