import torchio as tio


def get_train_transform():
    transform = tio.Compose([
        tio.RandomFlip(axes=('LR', 'AP', 'IS'), p=0.75),
        # tio.RandomAffine(scales=(0.9, 1.1), degrees=10, translation=2, p=0.5),
        tio.RandomNoise(std=0.01, p=0.3),
        tio.RandomGamma(log_gamma=True, p=0.3),
        tio.RandomBiasField(coefficients=0.3, p=0.2),
        tio.RandomBlur(std=(0.3, 0.7), p=0.2),
    ])
    return transform