def augment_linear_downsampling(data_sample, zoom_range=(0.5, 1), per_channel=True, p_per_channel=1,
channels=None, order_downsample=1, order_upsample=0, ignore_axes=None, p_per_sample=1):
if np.random.uniform() < p_per_sample:
augment_linear_downsampling_scipy(data_sample, zoom_range=zoom_range, per_channel=per_channel,
p_per_channel=p_per_channel, channels=channels, order_downsample=order_downsample,
order_upsample=order_upsample, ignore_axes=ignore_axes)
return data_sample
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_linear_downsampling_scipy(
data_dict[self.data_key][b],
zoom_range=self.zoom_range,
per_channel=self.per_channel,
p_per_channel=self.p_per_channel,
channels=self.channels,
order_downsample=self.order_downsample,
order_upsample=self.order_upsample)
return data_dict
def setUp(self):
np.random.seed(1234)
self.data_3D = np.random.random((2, 64, 56, 48))
self.data_2D = np.random.random((2, 64, 56))
self.data_3D_unique = np.reshape(range(2 * 64 * 56 * 48),
newshape=(2, 64, 56, 48))
self.data_2D_unique = np.reshape(range(2 * 64 * 56),
newshape=(2, 64, 56))
self.d_3D = augment_linear_downsampling_scipy(np.copy(self.data_3D),
zoom_range=[0.5, 1.5],
per_channel=False)
self.d_2D = augment_linear_downsampling_scipy(np.copy(self.data_2D),
zoom_range=[0.5, 1.5],
per_channel=False)
self.d_3D_channel = augment_linear_downsampling_scipy(
np.copy(self.data_3D),
zoom_range=[0.5, 1.5],
per_channel=False,
channels=[0])
self.d_2D_channel = augment_linear_downsampling_scipy(
np.copy(self.data_2D),
zoom_range=[0.5, 1.5],
per_channel=False,
channels=[0])
self.zoom_factor = 0.5
self.d_3D_upsample = augment_linear_downsampling_scipy(
np.copy(self.data_3D_unique),
zoom_range=[self.zoom_factor, self.zoom_factor],
per_channel=False,
order_downsample=0)
self.d_2D_upsample = augment_linear_downsampling_scipy(
np.copy(self.data_2D_unique),
zoom_range=[self.zoom_factor, self.zoom_factor],
per_channel=False,
order_downsample=0)
def __call__(self, **data_dict):
data_dict[self.data_key] = augment_linear_downsampling_scipy(data_dict[self.data_key], zoom_range=self.zoom_range)
return data_dict
def __call__(self, **data_dict):
data_dict['data'] = augment_linear_downsampling_scipy(data_dict['data'], zoom_range=self.zoom_range)
return data_dict
