matplotlib
scikit-image
scikit-learn
scipy
numpy
torch==1.4.0
torchvision==0.5.0
