Pillow
numpy
segmentation_models_pytorch
torch
torchvision
tensorflow