numpy
pillow
scipy
torch