| import albumentations as alb | |
| from albumentations.pytorch import ToTensorV2 | |
| import cv2 | |
| def get_augs(name): | |
| IMG_SIZE = 380 | |
| if name == "REAlbu": | |
| return alb.Compose([ | |
| alb.HorizontalFlip(), | |
| alb.CoarseDropout(max_holes = 1, min_height=int(IMG_SIZE*0.02), max_height=int(IMG_SIZE*0.2), min_width=int(IMG_SIZE*0.02), max_width=int(IMG_SIZE*0.2), p=1), | |
| ]) | |
| elif name == "RandCropAlbu": | |
| return alb.Compose([ | |
| alb.HorizontalFlip(), | |
| alb.RandomResizedCrop(height = IMG_SIZE, width = IMG_SIZE, scale=(1/1.3, 1.0), ratio=(0.9,1.1)), | |
| ]) | |
| elif name == "DFDCAlbu": | |
| return alb.Compose([ | |
| alb.ImageCompression(quality_lower=60, quality_upper=100, p=0.5), | |
| alb.GaussNoise(p=0.1), | |
| alb.GaussianBlur(blur_limit=3, p=0.05), | |
| alb.HorizontalFlip(), | |
| alb.OneOf([ | |
| alb.LongestMaxSize(max_size=IMG_SIZE, interpolation=cv2.INTER_CUBIC), | |
| alb.LongestMaxSize(max_size=IMG_SIZE, interpolation=cv2.INTER_AREA), | |
| alb.LongestMaxSize(max_size=IMG_SIZE, interpolation=cv2.INTER_LINEAR) | |
| ], p=1.0), | |
| alb.PadIfNeeded(min_height=IMG_SIZE, min_width=IMG_SIZE, border_mode=cv2.BORDER_CONSTANT), | |
| alb.OneOf([alb.RandomBrightnessContrast(), alb.FancyPCA(), alb.HueSaturationValue()], p=0.7), | |
| alb.ToGray(p=0.2), | |
| alb.ShiftScaleRotate(shift_limit=0.1, scale_limit=0.2, rotate_limit=10, border_mode=cv2.BORDER_CONSTANT, p=0.5), | |
| ]) |