如何通过TensorFlow-Slim VGG预先训练好的网络批量传递图片？

Question

我想通过网络传递图像以完成迁移学习任务。在下面的代码中，我将构建图形，然后获取完全连接层的输出。我想成批获得输出，因为我有一个包含超过20k图像的数组。

vgg.vgg_16(images)要求images是一个图像数组。我尝试输入占位符(在查看docs之后)，但是在加载检查点时，我得到了一个错误的There are no variables to save。

我可以一次给vgg.vgg_16(images)提供几张图片，但我需要为每一批加载检查点。我很确定有更好的方法可以做到这一点。有没有我可以参考的例子或参考资料？

from tensorflow.contrib import slim
from tensorflow.contrib.slim.nets import vgg

images = np.array(read_images(val_filenames[:4], 224, 224), dtype=np.float32) # load images and resize to 224 x 224


vgg_graph = tf.Graph()

with vgg_graph.as_default():
    with slim.arg_scope(vgg.vgg_arg_scope()):
        outputs, end_points = vgg.vgg_16(images, is_training=False)

    fc6 = end_points['vgg_16/fc6']


with tf.Session(graph=vgg_graph) as sess:
    saver = tf.train.Saver()
    saver.restore(sess, 'checkpoints/vgg_16.ckpt')

    # pass images through the network
    fc6_output = sess.run(fc6)

我还尝试了this和this参考资料，但我没有找到答案。

Answer 1

您可以创建一个可以传递给vgg网络的placeholder。将您的代码更改为：

images = tf.placeholder(tf.float32, shape=[batch_size, height, width, channels])

with slim.arg_scope(vgg.vgg_arg_scope()):
    outputs, end_points = vgg.vgg_16(images, is_training=False)

在训练期间，将输入传递给网络：

fc6_output = sess.run(fc6, feed_dict={images:batch_images})