pyyaml>=3.12
matplotlib
numpy
scikit-learn
scikit-image
torch
