src/util.py

import torchvision.transforms as transforms
import torch

# Define the transformation pipeline
transform = transforms.Compose([
    transforms.Resize((224,224)),
    transforms.ToTensor(),  # Convert the images to PyTorch tensors
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])