[bc8010]: / SegNet / SegNetCMR / inputs.py

Download this file

9 lines (6 with data), 272 Bytes

1
2
3
4
5
6
7
8
9
import tensorflow as tf
def placeholder_inputs(batch_size):
images = tf.placeholder(tf.float32, [batch_size, 256, 256, 1])
labels = tf.placeholder(tf.int64, [batch_size, 256, 256])
is_training = tf.placeholder(tf.bool)
return images, labels, is_training