torch==1.8.1
torchvision==0.9.1
pytorch-lightning==1.3.7
torchmetrics>=0.4.0
segmentation-models-pytorch
hydra-core==1.0.6
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

[tests]
pytest
scikit-image
parameterized
