深度学习提取特征向量

深度学习的特征向量可以用于图片检索,比较其特征向量相似的图片即是同类图片,提取神经网络中最后一层全连接层的特征向量的话只需要将分类层删掉即可,最后的输出即是该图片的特征向量。前提是有一个已经在自己的数据集上训练好的参数。
在测试开始之前加入:
y = network.outputs
在epoch时加入:
img_code=sess.run(y, feed_dict={x: X_test_a, y_: y_test_a})

你可能感兴趣的:(特征向量提取)