matplotlib
numpy
scipy
scikit-image
torch>=1.11.0
torchvision>=0.12.0
