Assuming the shape of x_in and y_in

[nathanin]

tfmodels/tfmodels/segmentation/segmentation_basemodel.py

Lines 58 to 60 in ae39722

	self.x_in = tf.placeholder_with_default(self.dataset.image_op,
	shape=[None, self.x_dims[0], self.x_dims[1], self.x_dims[2]],
	name='x_in')

tfmodels/tfmodels/segmentation/segmentation_basemodel.py

Lines 61 to 62 in ae39722

	self.y_in = tf.placeholder_with_default(self.dataset.mask_op,
	shape=[None, self.x_dims[0], self.x_dims[1], 1], name='y_in')