numpy
torch
Pillow
scikit-learn
scipy