torch==1.9.0
torchvision==0.10.0
pytorch-lightning==1.3.8
torchmetrics>=0.4.0
segmentation-models-pytorch>=0.2.0
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
