UNet featurizer is getting some dimention error using certain input data (e.g. skimage coin example). Should investigate this further.