setuptools
numpy
scikit-image
torch
torchvision
