numpy
matplotlib
tqdm
torch
torchmetrics
segmentation_models_pytorch