torch
Pillow
numpy
torchvision
