torch>=1.8.1
torchvision>=0.9.0
pytorch-lightning==1.3.8
torchmetrics>=0.4.0
segmentation-models-pytorch>=0.2.0
hydra-core==1.1.0
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

[tests]
pytest
scikit-image
parameterized
