torch
scipy
numpy
Pillow
torchvision
