torch>=1.8.1
torchvision>=0.9.0
pytorch-lightning==1.4.8
torchmetrics>=0.4.0
segmentation-models-pytorch>=0.2.0
hydra-core==1.1.1
kornia
albumentations
pandas
tensorboardX
pillow
matplotlib
scipy
numpy
pytorch_toolbelt
descartes
fiona
psycopg2
geopandas
geoalchemy2
rasterio
numba
skan
torch-scatter
tqdm
pygeos
rtree
bidict
Cython
ninja
pyyaml
pycocotools
multiprocess
wget
fastapi==0.68.1
uvicorn==0.15.0

[tests]
pytest
scikit-image
parameterized
