torchvision
numpy
torch
PIL