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