requirements.txt

# This file is managed by dependabot and should not be manually edited unless you need
# to add or remove a dependency. It documents the version of our dependencies used in
# CI, these are not hard requirements for using TorchGeo.

# setup
setuptools==62.3.2
wheel==0.37.1

# install
einops==0.4.1
fiona==1.8.21
kornia==0.6.5
matplotlib==3.5.2
numpy==1.22.4;python_version>='3.8'
numpy==1.21.6;python_version=='3.7'
omegaconf==2.2.2
pillow==9.1.1
pyproj==3.3.1
pytorch-lightning==1.6.4
rasterio==1.3b1
rtree==1.0.0
scikit-learn==1.1.1
segmentation-models-pytorch==0.2.1
shapely==1.8.2
timm==0.5.4
torch==1.11.0
torchmetrics==0.9.0
torchvision==0.12.0

# datasets
h5py==3.6.0
laspy==2.1.2
open3d==0.15.2;python_version<'3.10'
opencv-python==4.5.5.64
pandas==1.4.2
pycocotools==2.0.4
radiant-mlhub==0.5.1
rarfile==4.0
scipy==1.8.1
zipfile-deflate64==0.2.0

# style
black==22.3.0
flake8==4.0.1
isort[colors]==5.10.1
pydocstyle[toml]==6.1.1
pyupgrade==2.32.1

# tests
mypy==0.960
nbmake==1.3.0
pytest==7.1.2
pytest-cov==3.0.0

# docs
ipywidgets==7.7.0
nbsphinx==0.8.9
-e git+https://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme
sphinx==5.0.1