matplotlib
scikit-image
scikit-learn
scipy
numpy
torch
torchvision
