UNet featurizer error with some input

[marshuang80]

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