torch
torchvision
albumentations
opencv-python
pillow
numpy
gradio
segmentation_models_pytorch
matplotlib