无法使用tf-slim框架从mnist tfrecord获取图像。

Question

我使用DatasetDataProvider从tfrecord获取图像。我可以‘打印(图像)’，但当使用‘sess.run(图像)’来获取它时，程序似乎陷入了一个无限循环。我不知道我是否犯了错误。

打印(图像)

  Tensor("Reshape_3:0", shape=(28, 28, 1), dtype=uint8, device=/device:CPU:0)

完整代码如下：

    from __future__ import absolute_import
    from __future__ import division
    from __future__ import print_function

    import tensorflow as tf

    from datasets import dataset_factory
    from tensorflow.contrib import slim

    dataset = dataset_factory.get_dataset(
        'mnist', 'train', '/home/zehao/Dataset/mnist')

    with tf.device('/cpu:0'):
        provider = slim.dataset_data_provider.DatasetDataProvider(
            dataset,
            num_readers=1,
            common_queue_capacity=20 * 1,
            common_queue_min=10 * 1)
        [image, label] = provider.get(['image', 'label'])

    print(image)

    sess = tf.Session()
    sess.run(image)

Answer 1

slim.dataset_data_provider在引擎盖下使用TensorFlow 输入队列。因此，重要的是(在创建会话之后)添加以下2行来启动队列运行程序：

coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(coord=coord)

有关完整的示例，请参见：数据