torch>2.0
torchvision
rioxarray
albumentations==1.4.10
albucore==0.0.16
rasterio
torchmetrics
geopandas
lightly
jsonargparse<=4.35.0
h5py
hdf5plugin
numba
mlflow
lightning
segmentation-models-pytorch<=0.4
jsonargparse
pytest
torchgeo>=0.7.0
einops
timm>=1.0.15
pycocotools
huggingface_hub
tifffile
python-box
wandb
termcolor
numba
hdf5plugin
h5netcdf
pytest-sugar

[dev]
black
mkdocs-material
mkdocstrings[python]
mike
mkdocs-git-revision-date-localized-plugin
pre-commit

[mmseg]
mmcv
ftfy
regex
openmim

[peft]
peft>=0.15.0

[test]
pytest
pytest-coverage
coverage-badge
peft>=0.15.0
diffusers<=0.34.0
tokenizers
dask
wandb

[visualize]
torchview
graphviz

[wxc]
PrithviWxC
granitewxc
