numpy
torch
torchvision
segmentation-models-pytorch>=0.2.0
monai[ignite,itk,matplotlib,mlflow,pandas,pillow,scipy,tensorboard,tifffile,tqdm]
torchmetrics
tifffile
imagecodecs
tqdm

[all]
AM-model-training[reporting,scripts]

[cpu]
torch>=2.7.0
torchvision>=0.22.0

[cu113]
torch>=2.7.0
torchvision>=0.22.0

[cu118]
torch>=2.7.0
torchvision>=0.22.0

[cu126]
torch>=2.7.0
torchvision>=0.22.0

[cu128]
torch>=2.7.0
torchvision>=0.22.0

[reporting]
mlflow

[scripts]
jupyter
ipywidgets
