Pillow
numpy
matplotlib
json
torch
torchvision
