torch
torchvision
segmentation-models-pytorch