def __call__(self, **data_dict):
data = data_dict.get(self.data_key)
ret_val = augment_channel_translation(data=data, const_channel=self.const_channel, max_shifts=self.max_shift)
data_dict[self.data_key] = ret_val[0]
return data_dict
def __call__(self, **data_dict):
data = data_dict.get(self.data_key)
ret_val = augment_channel_translation(data=data, const_channel=self.const_channel, max_shifts=self.max_shift)
data_dict[self.data_key] = ret_val[0]
return data_dict
def channel_translation_generator(generator, const_channel=0, max_shifts=None):
"""
Translates all channels within an instance of a batch according to randomly drawn shifts from within [-max_shift, max_shift].
One channel is held constant, the others are shifted in the same manner.
:param generator:
:param const_channel:
:param max_shifts:
:return:
"""
if max_shifts is None:
max_shifts = {'z': 2, 'y': 2, 'x': 2}
for data_dict in generator:
data_dict["data"] = augment_channel_translation(data_dict["data"], const_channel, max_shifts)
yield data_dict
def channel_translation_generator(generator, const_channel=0, max_shifts=None):
"""
Translates all channels within an instance of a batch according to randomly drawn shifts from within [-max_shift, max_shift].
One channel is held constant, the others are shifted in the same manner.
:param generator:
:param const_channel:
:param max_shifts:
:return:
"""
if max_shifts is None:
max_shifts = {'z': 2, 'y': 2, 'x': 2}
for data_dict in generator:
data_dict["data"] = augment_channel_translation(
data_dict["data"], const_channel, max_shifts)
yield data_dict