Utilities/transforms.py

import albumentations as A
from albumentations.pytorch import ToTensorV2

# Test data transformations
test_transforms = A.Compose(
    [
        A.Normalize((0.4914, 0.4822, 0.4465), (0.247, 0.243, 0.261)),
        ToTensorV2(),
    ]
)