--- a +++ b/SegNet/SegNetCMR/inputs.py @@ -0,0 +1,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 \ No newline at end of file