Preprocess.py

import SimpleITK as sitk
import numpy as np

def standardize_MRI(image: sitk.Image) -> sitk.Image:    
    data = sitk.GetArrayFromImage(image)
    data = (data-data.mean())/data.std()
    result = sitk.GetImageFromArray(data)
    result.CopyInformation(image)
    return result

def clip_and_standardize_CT(image: sitk.Image) -> sitk.Image:    
    data = sitk.GetArrayFromImage(image)
    data[data < -1024] = -1024
    data[data > 276.0] = 276
    data = (data-(-370.00039267657144))/436.5998675471528
    result = sitk.GetImageFromArray(data)
    result.CopyInformation(image)
    return result

def standardize_ImageNet(image: sitk.Image) -> sitk.Image:
    data = sitk.GetArrayFromImage(image)
    data = (data-np.min(data))/(np.max(data)-np.min(data))
    data = (data-0.485)/0.229
    result = sitk.GetImageFromArray(data)
    result.CopyInformation(image)
    return result