albumentations
matplotlib
segmentation_models.pytorch
torch
pytorch-lightning
