前几篇博客写了如何处理数据,如何把用自己的数据训练VGG-16,如何把训练好的模型保存。而在实际应用中,并不是所有的操作都是为了分类的,有时候需要提取图像的特征,那么怎么利用已经保存的模型提取特征呢?
“桃叶儿尖上尖,柳叶儿就遮满了天”
测试数据转换成tfrecords,教程:点击打开链接
保存训练好的VGG-16模型,教程:点击打开链接
1、读取测试数据
首先把测试数据转换成tfrecords,然后读取出来,代码和前面博客写的一致:
#读取文件
def read_and_decode(filename,batch_size):
#根据文件名生成一个队列
filename_queue = tf.train.string