def configure_transforms(resize_to): train_transforms = iaa.Sequential([ iaa.PadToAspectRatio(1.0, position='center'), iaa.Resize({ "height": resize_to, "width": resize_to }), iaa.Affine( scale={ "x": (0.95, 1.05), "y": (0.95, 1.05) }, translate_percent={ "x": (-0.05, 0.05), "y": (-0.05, 0.05) }, rotate=(-10, 10), shear=(-5, 5), ), ]) valid_transforms = iaa.Sequential([ iaa.PadToAspectRatio(1.0, position='center'), iaa.Resize({ "height": resize_to, "width": resize_to }), ]) return train_transforms, valid_transforms
def __init__(self, ): self.augmentations = iaa.Sequential([ iaa.PadToAspectRatio( 1.0, position="center-center").to_deterministic() ])
ocr_synthetic_transforms = iaa.Sequential([ iaa.Resize({ 'height': 64, 'width': 320 }), iaa.PerspectiveTransform(scale=(0.01, 0.05)), iaa.Sometimes(0.5, iaa.Multiply((0.5, 1.5))), iaa.Sometimes(0.5, iaa.JpegCompression(compression=(70, 99))), iaa.Affine( scale={ 'x': (0.95, 1.02), 'y': (0.95, 1.02) }, translate_percent={ 'x': (-0.02, 0.02), 'y': (-0.02, 0.02) }, rotate=(-3, 3), shear=(-5, 5), ) ]) pad_transform = iaa.PadToAspectRatio(1.0, position='center') ocr_transforms = iaa.Sequential([ iaa.Resize({ "height": 64, "width": 320 }), ])
## Pad elif augmentation == 'pad_to_fixed_size': transform = iaa.PadToFixedSize(width=1000, height=1000) transformed_image = transform(image=image) elif augmentation == 'pad_to_multiples_of': transform = iaa.PadToPowersOf(height_base=3, width_base=2) transformed_image = transform(image=image) elif augmentation == 'pad_to_powers_of': transform = iaa.CropToMultiplesOf(height_multiple=32, width_multiple=32) transformed_image = transform(image=image) elif augmentation == 'pad_to_aspect_ratio': transform = iaa.PadToAspectRatio(2.0) transformed_image = transform(image=image) elif augmentation == 'pad_to_square': transform = Resize(always_apply=True, height=200, width=400) transformed_image = transform(image=image)['image'] transform = iaa.PadToSquare() transformed_image = transform(image=transformed_image) elif augmentation == 'center_pad_to_fixed_size': transform = iaa.CenterPadToFixedSize(width=1000, height=1000) transformed_image = transform(image=image) elif augmentation == 'center_pad_to_multiples_of': transform = iaa.CenterPadToPowersOf(height_base=3, width_base=2) transformed_image = transform(image=image)