torchvision
numpy
kornia
wandb
omegaconf
matplotlib
rasterio
scikit-image
Pillow
huggingface-hub

[:python_version < "3.11"]
torch==1.13.1
pytorch-lightning==1.9.*

[:python_version >= "3.12"]
torch>=2.1
pytorch-lightning<3.0,>=2.1

[docs]
mkdocs-material

[tests]
pytest
pytest-cov
numpy
