基于图像自编码的神经网络特征学习方法
摘要:
本发明公开了基于图像自编码的神经网络特征学习方法,属于特征学习和图像检索技术领域。首先通过多标签图像数据集的分割标签构造训练图像集对应的分割训练图像集,然后初始化卷积神经网络和自编码神经网络的权重,使用随机梯度下降法训练自编码神经网络,提取每张训练样本对应的分割图像的隐含变量并进行归一化。随后,使用该隐含变量作为训练集原始图像对应的训练目标,训练卷积神经网络,并提取测试集合图像库中的每幅图像对应的特征向量,通过计算查询图像和图像库中每幅图像的特征向量间的欧式距离,并将距离按从小到大的顺序排列,得到相似图像检索结果。本发明使训练出的神经网络提取的特征在多标签检索任务上取得了更加优秀的检索效果。
公开/授权文献
0/0