利索能及
我要发布
收藏
专利号: 2022106512648
申请人: 电子科技大学中山学院
专利类型:发明专利
专利状态:已下证
更新日期:2026-06-16
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种图像类别预测方法,其特征在于,包括:获取待处理图像,并对所述待处理图像进行随机裁剪,获得局部图像;

分别提取出所述待处理图像中的全局特征和所述局部图像中的局部特征;

使用图神经网络对所述全局特征和所述局部特征进行聚合,获得所述待处理图像的聚合特征;

根据所述待处理图像的聚合特征与所述图神经网络中的类原型聚合特征之间的相似度预测所述待处理图像的类别,所述类原型聚合特征是使用所述图神经网络对已知类别标签对应的样本图像训练获得的。

2.根据权利要求1所述的方法,其特征在于,所述使用图神经网络对所述全局特征和所述局部特征进行聚合,获得所述待处理图像的聚合特征,包括:获取所述全局特征在所述图神经网络中的待处理全局节点,以及所述局部特征在所述图神经网络中的待处理局部节点;

提取出所述待处理全局节点与所述待处理局部节点之间的待处理边特征;

对所述待处理全局节点、所述待处理局部节点和所述待处理边特征进行聚合处理,获得所述待处理图像的聚合特征。

3.根据权利要求1所述的方法,其特征在于,在所述使用图神经网络对所述全局特征和所述局部特征进行聚合之前,还包括:获取多个样本图像和多个样本标签,所述样本标签包括所述样本图像的类别标签;

使用所述多个样本图像和所述多个样本标签构建初始图结构;

以所述多个样本图像为训练数据,以所述多个样本标签为训练标签,对所述初始图结构进行训练,获得所述图神经网络。

4.根据权利要求3所述的方法,其特征在于,所述使用所述多个样本图像和所述多个样本标签构建初始图结构,包括:对所述多个样本图像中的每个样本图像进行随机裁剪,获得多个样本局部图像;

分别提取出所述多个样本图像中的样本全局特征和所述多个样本局部图像中的样本局部特征;

使用所述多个样本图像中的样本全局特征和所述多个样本局部图像中的样本局部特征构建所述初始图结构。

5.根据权利要求4所述的方法,其特征在于,所述使用所述多个样本图像中的样本全局特征和所述多个样本局部图像中的样本局部特征构建所述初始图结构,包括:以所述多个样本图像中的样本全局特征为全局节点,并将所述多个样本图像对应的全局节点相互连接,获得全连接图;

以所述多个样本局部图像中的样本局部特征为局部节点,并将该局部节点与所述全连接图中对应的全局节点相互连接,获得所述初始图结构。

6.根据权利要求3所述的方法,其特征在于,所述对所述初始图结构进行训练,包括:对所述初始图结构进行聚合和平均处理,获得所述类原型的聚合特征;

使用所述初始图结构提取出所述样本图像的聚合特征,并计算所述样本图像的聚合特征与所述类原型的聚合特征之间的分类损失值;

使用所述初始图结构提取出所述样本图像的边特征,并计算所述样本图像的边特征与所述样本标签计算出的边特征之间的边损失值;

根据所述分类损失值和所述边损失值对所述初始图结构进行训练。

7.根据权利要求6所述的方法,其特征在于,所述对所述初始图结构进行聚合和平均处理,获得所述类原型的聚合特征,包括:分别计算出所述初始图结构中的全局节点与全局节点之间的第一边特征,以及所述初始图结构中的全局节点与局部节点之间的第二边特征;

针对所述初始图结构中的每个全局节点,对该全局节点对应的样本全局特征、第一边特征和第二边特征进行聚合和特征转换更新,获得全局聚合特征;

针对所述初始图结构中的每个局部节点,对该局部节点对应的样本局部特征和第二边特征进行聚合和特征转换更新,获得局部聚合特征;

将每个类别对应的所述全局聚合特征和所述局部聚合特征进行平均处理,获得所述类原型的聚合特征。

8.一种图像类别预测装置,其特征在于,包括:图像获取裁剪模块,用于获取待处理图像,并对所述待处理图像进行随机裁剪,获得局部图像;

图像特征提取模块,用于分别提取出所述待处理图像中的全局特征和所述局部图像中的局部特征;

聚合特征获得模块,用于使用图神经网络对所述全局特征和所述局部特征进行聚合,获得所述待处理图像的聚合特征;

图像类别预测模块,用于根据所述待处理图像的聚合特征与所述图神经网络中的类原型聚合特征之间的相似度预测所述待处理图像的类别,所述类原型聚合特征是使用所述图神经网络对已知类别标签对应的样本图像训练获得的。

9.一种电子设备,其特征在于,包括:处理器和存储器,所述存储器存储有所述处理器可执行的机器可读指令,所述机器可读指令被所述处理器执行时执行如权利要求1至7任一所述的方法。

10.一种计算机可读存储介质,其特征在于,该计算机可读存储介质上存储有计算机程序,该计算机程序被处理器运行时执行如权利要求1至7任一所述的方法。