lightning
numpy
segmentation-models-pytorch
torch
torchmetrics
torchvision