ColaWithIce

Here is my code: ``` def toGray(image, **params): gray = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY) return gray transform_A = A.ReplayCompose([ A.Lambda(name='togray',image=toGray), A.Resize(256,256), A.HorizontalFlip(p=0.5), A.ShiftScaleRotate(shift_limit=0.1, scale_limit=0.2, rotate_limit=30, p=0.5, border_mode=cv2.BORDER_CONSTANT), A.RandomBrightnessContrast(p=0.5), A.Normalize((0.5,),(0.5,), always_apply=True), ToTensorV2() ])...