torch
torchvision
opencv-python
Pillow
segmentation-models-pytorch==0.3.3
