def data_prep_function_valid(x, p_transform=p_transform, **kwargs):
#take a patch in the middle of the chip
x = x.astype(np.float32)
x = data_transforms.channel_zmuv(x,
img_stats=channel_zmuv_stats,
no_channels=4)
return x
def data_prep_function_train(x,
p_transform=p_transform,
p_augmentation=p_augmentation,
**kwargs):
x = np.array(x, dtype=np.float32)
x = data_transforms.channel_zmuv(x,
img_stats=channel_zmuv_stats,
no_channels=4)
x = data_transforms.random_lossless(x, p_augmentation, rng)
return x
def data_prep_function_train(x,
p_transform=p_transform,
p_augmentation=p_augmentation,
**kwargs):
x = x.astype(np.float32)
x = data_transforms.perturb(x, p_augmentation, p_transform['patch_size'],
rng)
x = data_transforms.channel_zmuv(x,
img_stats=channel_zmuv_stats,
no_channels=4)
return x
def data_prep_function_valid(x, p_transform=p_transform, **kwargs):
x = np.array(x, dtype=np.float32)
x = data_transforms.channel_zmuv(x,
img_stats=channel_zmuv_stats,
no_channels=4)
return x
