def transforms(self):
composed = transforms.Compose([
tfs.Rescale(out_size=self.rescale_shape),
tfs.ReCrop(out_size = (self.nn_input_image_shape)),
tfs.Rotate(rotations = self.rotation_angles,rotation_probabilities = self.rotation_angle_probabilities),
tfs.Flip(flip_probabilites = self.flip_hor_ver_probabilities),
tfs.Apply_Gaussian_filter(probabity=self.gaussian_filter_probability),
tfs.Apply_Unsharpmask_filter(probabity=self.unsharpmask_filter_probability)
])
return composed
def __init__(self, transformation_parameters):
self.nn_input_image_shape = transformation_parameters['nn_input_image_shape']
crop_pixel_count = transformation_parameters['crop_pixel']
self.rescale_shape = [x+y for x,y in zip(self.nn_input_image_shape,crop_pixel_count)]
self.rotation_angles = transformation_parameters['rotation_angles']
self.rotation_angle_probabilities = transformation_parameters['rotaion_angle_probabilities']
self.flip_hor_ver_probabilities = transformation_parameters['flip_hor_ver_probabilities']
self.gaussian_filter_probability = transformation_parameters['gaussian_filter_probability']
self.unsharpmask_filter_probability = transformation_parameters['unsharp_mask_filter_probability']
self.Rescaler = tfs.Rescale(out_size=self.rescale_shape)
self.ReCropper = tfs.ReCrop(out_size = (self.nn_input_image_shape))
self.Rotator = tfs.Rotate(rotations = self.rotation_angles,
rotation_probabilities = self.rotation_angle_probabilities)
self.Flipper = tfs.Flip(flip_probabilites = self.flip_hor_ver_probabilities)
self.Gaussian_filter = tfs.Apply_Gaussian_filter(probabity=self.gaussian_filter_probability)
self.Unsharpmask_filter = tfs.Apply_Unsharpmask_filter(probabity=self.unsharpmask_filter_probability)
