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

Tensorflow 无法使用tf slim框架从mnist tfrecord获取图像,tensorflow,Tensorflow,我使用DatasetDataProvider从tfrecord获取图像。我可以“打印(图像)”,但当使用“sess.run(图像)”获取它时,程序似乎陷入了无限循环。我不知道我是否犯了错误 打印(图像)获取 完整代码如下: from __future__ import absolute_import from __future__ import division from __future__ import print_function import tenso

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

打印(图像)获取

完整代码如下:

    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)

slim.dataset\u data\u提供程序
在引擎盖下使用TensorFlow。因此,(在创建会话后)添加以下两行以启动队列运行程序非常重要:

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

有关完整示例,请参见:

slim.dataset\u data\u提供程序在引擎盖下使用TensorFlow。因此,(在创建会话后)添加以下两行以启动队列运行程序非常重要:

coord = tf.train.Coordinator()
threads = tf.train.start_queue_runners(coord=coord)
有关完整示例,请参见: