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)
