numpy
Pillow
tensorboardX
torch
torchvision