基于图卷积网络的视觉检索模型的无监督训练方法和装置
摘要:
本发明提供一种基于图卷积网络的视觉检索模型的无监督训练方法和装置。其中方法包括:将图像样本集输入特征提取网络得到特征提取网络输出的目标特征信息;基于目标特征信息构建图像样本集的无向图,将无向图输入图卷积网络,将图像样本集输入特征提取网络,获取特征提取网络的中间层特征信息输入图卷积网络,得到图卷积网络输出第一概率分布信息和特征提取网络输出的目标特征信息;基于目标特征信息对图像样本集进行聚类,基于聚类的结果确定图像样本集的第二概率分布信息,基于第一概率分布信息和第二概率分布信息对特征提取网络和图卷积网络的参数进行调整。本发明可以提升样本的特征表达,从而提升特征提取网络的性能,提升视觉检索的效果。
0/0