pillow
torch
torchvision