def get_dataloader()

in utils.py [0:0]

• 9 lines of code
• 1 McCabe index (conditional complexity)

def get_dataloader():
    val_transform = transforms.Compose([
        transforms.Resize((128, 128)),
        transforms.ToTensor(),
        transforms.Normalize(image_norm_mean, image_norm_stddev),
    ])
    dataset = datasets.ImageFolder('GTSRB/Final_Test/', val_transform)
    val_dataloader = torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=True)

    return val_dataloader