一 实例描述
在with tf.Session函数中加入启动队列,并通过加入coord协调器的方式使session close时同步内部线程一起退出。
二 代码
import cifar10_input import tensorflow as tf import pylab #取数据 batch_size = 12 data_dir = '/tmp/cifar10_data/cifar-10-batches-bin' images_test, labels_test = cifar10_input.inputs(eval_data = True, data_dir = data_dir, batch_size = batch_size) with tf.Session() as sess: tf.global_variables_initializer().run() #定义协调器 coord = tf.train.Coordinator() threads = tf.train.start_queue_runners(sess, coord) image_batch, label_batch = sess.run([images_test, labels_test]) print("__\n",image_batch[0]) print("__\n",label_batch[0]) pylab.imshow(image_batch[0]) pylab.show() coord.request_stop()
三 运行结果