torch
torchvision
Pillow==5.2.0
numpy
argparse