torch
numpy
torchvision
pillow
