记一次TensorFlow tf.summary.image对生成batch数据的影响
Published:
记一次TensorFlow tf.summary.image对生成batch数据的影响
最近在尝试用TensorFlow标准文件格式tfrecords和内置的文件队列向模型读入数据。 首先分别创建train和test两个文件队列,读入mnist数据,生成train和test的两个batch的tensor: img_batch,label_batch和test_img_batch, test_label_batch。然后用tf.summary.image()在tensorboard的IMAGES栏里显示图片。最后在sess里用两个sess分别生成以batch的训练数据和测试数据。代码如下: ``` train_queue = tf.train.string_input_producer([train_tfrecords]) test_queue = tf.train.string_input_producer([test_tfrecords]) img, label = mnist.read_and_decode(train_queue) test_img, test_label = mnist.read_and_decode(test_queue) img_batch, label_batch = tf.train.batch([img, label], batch_size=2, capacity=20) test_img_batch, test_label_batch = tf.train.batch([test_img, test_label], batch_size=2, capacity=20)