torch
torchvision
pytorch-lightning==1.1.8
segmentation-models-pytorch
hydra-core
kornia
albumentations
pandas
tensorboardX
pillow
matplotlib
scipy
numpy
pytorch_toolbelt
descartes
fiona
psycopg2
geopandas
geoalchemy2
rasterio
numba
skan
torch-scatter
tqdm
pygeos
rtree
bidict

[tests]
pytest
scikit-image
parameterized
