def setUp(self):
np.random.seed(1234)
self.data_input_3D = np.random.random((2, 64, 56, 48))
self.data_input_2D = np.random.random((2, 64, 56))
self.factor = (0.75, 1.25)
self.multiplier_range = [2,4]
self.d_3D_per_channel = augment_brightness_additive(np.copy(self.data_input_3D), mu=100, sigma=10,
per_channel=True)
self.d_3D = augment_brightness_additive(np.copy(self.data_input_3D), mu=100, sigma=10, per_channel=False)
self.d_2D_per_channel = augment_brightness_additive(np.copy(self.data_input_2D), mu=100, sigma=10,
per_channel=True)
self.d_2D = augment_brightness_additive(np.copy(self.data_input_2D), mu=100, sigma=10, per_channel=False)
self.d_3D_per_channel_mult = augment_brightness_multiplicative(np.copy(self.data_input_3D),
multiplier_range=self.multiplier_range,
per_channel=True)
self.d_3D_mult = augment_brightness_multiplicative(np.copy(self.data_input_3D),
multiplier_range=self.multiplier_range, per_channel=False)
self.d_2D_per_channel_mult = augment_brightness_multiplicative(np.copy(self.data_input_2D),
multiplier_range=self.multiplier_range,
per_channel=True)
self.d_2D_mult = augment_brightness_multiplicative(np.copy(self.data_input_2D),
multiplier_range=self.multiplier_range, per_channel=False)
def __call__(self, **data_dict):
for b in range(len(data_dict[self.data_key])):
if np.random.uniform() < self.p_per_sample:
data_dict[self.data_key][b] = augment_brightness_multiplicative(data_dict[self.data_key][b],
self.multiplier_range,
self.per_channel)
return data_dict
def brightness_augmentation_by_multiplication_generator(generator, multiplier_range=(0.5, 2), per_channel=True):
warn("using deprecated generator brightness_augmentation_by_multiplication_generator", Warning)
'''
Multiplies each voxel with a randomly sampled multiplier.
This is done separately for each channel if per_channel is set to True.
'''
for data_dict in generator:
data_dict['data'] = augment_brightness_multiplicative(data_dict['data'], multiplier_range, per_channel)
yield data_dict
def __call__(self, **data_dict):
data_dict['data'] = augment_brightness_multiplicative(
data_dict['data'], self.multiplier_range, self.per_channel)
return data_dict
def __call__(self, **data_dict):
data_dict[self.data_key] = augment_brightness_multiplicative(data_dict[self.data_key], self.multiplier_range,
self.per_channel)
return data_dict
