|
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), |
|
]) |