image size is hard-coded..

[hvgazula]

tissue_labeling/TissueLabeling/data/dataset.py

Lines 59 to 61 in 59f5ea9

	if self.model_name == 'simple_unet':
	image = image[:,1:161,1:193]
	mask = mask[:,1:161,1:193]

Can we, not hard code numbers like this? Is there a better alternative?

[hvgazula]

time to update the unet and other models to accommodate images of varying size. For more info, please refer to links in #44