albumentations
matplotlib
numpy==1.26.4
segmentation_models.pytorch
torch
pytorch-lightning
