Tensorflow 无法使用tf slim框架从mnist tfrecord获取图像
我使用DatasetDataProvider从tfrecord获取图像。我可以“打印(图像)”,但当使用“sess.run(图像)”获取它时,程序似乎陷入了无限循环。我不知道我是否犯了错误 打印(图像)获取 完整代码如下: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
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)
有关完整示例,请参见: