Augmentation issues

[jesusbft]

The actual augmentation parameters have issues:

The ElasticTransform parameters makes the image unreadable
ElasticTransform(alpha=64, sigma=25, p=1.0),

The Affine and ShiftScaleRotate options cut the borders and lose informations:
Affine(translate_px=(0, 5), rotate=(-3, 3), shear=(-5, 5), p=1.0)
ShiftScaleRotate(shift_limit=0.0625, scale_limit=0.2, rotate_limit=1, p=0.2) Similar

My suggestions are:

• ElasticTransform reduce the alpha parameter to equal/lower than 10 (I suggest the int 7)

• Substitute the Affine and ShiftScaleRotate to option SafeRotate, that don't remove any information of the image.

self._transforms = Compose([
                                    ToFloat(),
                                    PixelDropout(p=0.2),
                                    OneOf([
                                        MotionBlur(p=0.2),
                                        MedianBlur(blur_limit=3, p=0.1),
                                        Blur(blur_limit=3, p=0.1),
                                    ], p=0.2),
                                    OneOf([
                                        OpticalDistortion(p=0.3),
                                        ElasticTransform(alpha=7, sigma=25, p=0.1),
                                        SafeRotate(limit=(-3,3), border_mode=cv2.BORDER_CONSTANT, p=0.2)
                                    ], p=0.2),
                                   ], p=1.0)