Test torchvision tranforms on a bunch of images

test_transforms.py

from torchvision import datasets, models, transforms
from PIL import Image
import torch
import matplotlib.pyplot as plt
from pathlib import Path

images = list(Path('../foo').glob('**/*.*'))
example = images.pop()

img = Image.open(example).convert('RGB')
transform = transforms.ColorJitter(brightness=0.5, contrast=0.4, saturation=0.9, hue=0.15)

plt.subplots(3, 3)
ax = plt.gca()
for i in range(3):
    for i in range(3):
        transformed = transform(img)
        ax[i, j].imshow(transformed)
        plt.show()