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)