为seesion中的队列加上协调器

一 实例描述
在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()
三 运行结果

猜你喜欢

转载自blog.csdn.net/chengqiuming/article/details/80297123