torch
torchvision
Pillow
matplotlib
numpy
scipy