Creating comparable pipeline with DALI

AI & Data Science Deep Learning (Training & Inference) Frameworks
1

Hey there everybody!

I’m super new to this space, and I’m just starting to explore DALI as a means of speeding up my image classification model training. I’m realizing some pretty serious performance improvements, but I’m having trouble replicating the “accuracy” of my model that uses TorchVision to transform the images in my pipeline.

Does anyone know of a good resource that I could use to (more or less) replicate the following TorchVision pipeline using DALI?

transform = transforms.Compose([
    transforms.Resize((224, 224)),
    transforms.RandomHorizontalFlip(p=0.5),
    transforms.ColorJitter(brightness=0.2, contrast=0.2, saturation=0.2, hue=0.2),
    transforms.RandomAffine(degrees=30, translate=(0.1, 0.1), scale=(0.8, 1.2), shear=10),
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])

Thanks everyone!