matplotlib
numpy
pandas
Pillow
scikit_learn
torch
torchvision
