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