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

[docs]
mkdocs-material

[tests]
pytest
pytest-cov
numpy
