venv with python3.7 (install python with pyenv if it is not available and you have no sudo permissions)

pip install typing-extensions==4.7.1

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 -f <https://download.pytorch.org/whl/torch_stable.html>
pip install torch_scatter==2.0.9
pip install torch_geometric==1.7.2
pip install spconv-cu114==2.1.25
pip install torch_sparse==0.6.12 cumm-cu114==0.2.8 torch_cluster==1.5.9
pip install safetensors==0.3.3
pip install tensorboard timm termcolor tensorboardX

cd third_party/SparseTransformer && python [setup.py](<http://setup.py/>) install

If building torch_scatter or SparseTransformer fails (tested in thor):

export CC=gcc-10
export CXX=g++-10
export CUDAHOSTCXX=/usr/bin/g++-10
export TORCH_CUDA_ARCH_LIST="8.6"

pip install -e repos/DetectionMetrics/

pyproject for sphereformer:

[tool.poetry]
name = "detectionmetrics"
version = "2.2.0"
description = "Tools for evaluating segmentation and object detection models"
authors = ["JdeRobot", "d.pascualhe <[email protected]>"]
readme = "README.md"
license = "LICENSE"

[tool.poetry.dependencies]
python = "^3.7"
torch = "1.8.0"
pytorch-lightning = "1.3.8"
torchmetrics = "0.3.2"
tqdm = "^4.65.0"
pandas = "^1.3.5"
PyYAML = "^5.4.1"
pyarrow = "^12.0.1"
pillow = "9.5.0"
numpy = "1.21.6"
opencv-python-headless = "^4.5.5.64"
scikit-learn = "^1.0.2"
open3d = "^0.13.0"  # 0.17+ requires Python >=3.8
addict = "^2.4.0"
matplotlib = "^3.5.0"
click = "^8.0.4"
tensorboard = "^2.6.0"  # Newer versions require Python >=3.8

[tool.poetry.group.dev.dependencies]
black = "^22.3.0"
pylint = "^2.15.0"
ipykernel = "^6.9.1"

[tool.poetry.group.docs.dependencies]
sphinx = "^4.5.0"
sphinx-rtd-theme = "^1.0.0"

[tool.poetry.group.test.dependencies]
pytest = "^6.2.5"

[tool.poetry.scripts]
dm_evaluate = "detectionmetrics.cli.evaluate:evaluate"
dm_batch = "detectionmetrics.cli.batch:batch"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"