torch>=2.1
pytorch-lightning<3.0,>=2.1
torchvision
numpy
kornia
wandb
omegaconf
matplotlib
rasterio
scikit-image
Pillow
huggingface-hub

[docs]
mkdocs-material

[tests]
pytest
pytest-cov
numpy
