DALI-Post-1.1

dali-1.1.py

from torchvision import transforms

def get_image_transforms() -> transforms.Compose:
    """
    These transformations meant for data augmentation are a bottleneck
    since all the operations are done on CPU and then the tensors are
    copied to the GPU device.
    """
    return transforms.Compose([
        transforms.RandomSizedCrop(224),
        transforms.RandomHorizontalFlip(),
        transforms.ToTensor(),
        transforms.Normalize(mean=[0.485, 0.456, 0.406],
                             std=[0.229, 0.224, 0.225])
    ])