numpy
scikit-learn
scikit-image
Pillow
torch
torchvision
segmentation-models-pytorch
