class Normalization(object):
"""
Normalize an image to 0 - 255
"""
def __init__(self):
self.name = 'Normalization'
def __call__(self, sample):
resacleFilter = sitk.RescaleIntensityImageFilter()
resacleFilter.SetOutputMaximum(255)
resacleFilter.SetOutputMinimum(0)
image, label = sample['image'], sample['label']
image = resacleFilter.Execute(image)
return {'image': image, 'label': label}
sitk.RescaleIntensityImageFilter()对输入图像的强度级别应用线性变换