torch>=1.7.0
torchvision>=0.8.1
pytorch-lightning>=1.0.0
scikit-image>=0.17.2
numpy>=1.19.2
