MiniDPVO / pixi.toml
pablovela5620's picture
chore: Update post-install command to include timing information
2e45a07
raw
history blame
2.24 kB
[project]
name = "mini-dpvo"
version = "0.1.0"
description = "Add a short description here"
authors = ["pablovela5620 <pablovela5620@gmail.com>"]
channels = ["nvidia/label/cuda-11.8.0", "nvidia", "conda-forge", "pytorch", "pyg"]
platforms = ["linux-64"]
[system-requirements]
libc = { family="glibc", version="2.31" }
[tasks]
download-model = """
test -e checkpoints/dpvo.pth
|| (
wget https://www.dropbox.com/s/nap0u8zslspdwm4/models.zip
&& unzip models.zip -d checkpoints
&& rm -r models.zip
)
"""
download-dpvo-data = """
test -e data/movies/IMG_0492.MOV
|| (
wget https://www.dropbox.com/s/7030y0mdl6efteg/movies.zip -P data/
&& unzip data/movies.zip -d data/
&& rm -r data/movies.zip
)
"""
download-iphone-data = """
test -e data/iphone/pool.MOV
|| (
huggingface-cli download pablovela5620/dpvo-example-data pool.MOV --repo-type dataset --local-dir data/iphone/
)
"""
post-install = {cmd="time python -m pip install -e .", depends_on=["download-model", "download-dpvo-data", "download-iphone-data"], outputs=["cuda_ba.cpython-311-x86_64-linux-gnu.so"]}
rr-viewer = "rerun --memory-limit 50% --drop-at-latency 500ms"
demo = """
python tools/demo.py --imagedir data/movies/IMG_0493.MOV --config config/fast.yaml
"""
app = {cmd="python tools/app.py", depends_on=["post-install"]}
# Docker tasks
docker-build = "docker build --no-cache -t mini-dpvo ."
docker-run = "docker run --gpus all -it -p 7860:7860 mini-dpvo"
[dependencies]
python = "3.11.*"
pip = ">=24.0,<25"
cuda = {version = "*", channel="nvidia/label/cuda-11.8.0"}
pytorch-cuda = {version = "11.8.*", channel="pytorch"}
pytorch = {version = ">=2.2.0,<2.3", channel="pytorch"}
torchvision = {version = ">=0.17.0,<0.18", channel="pytorch"}
plyfile = ">=1.0.3,<1.1"
pytorch-scatter = {version="*", channel="pyg"}
matplotlib = ">=3.8.4,<3.9"
yacs = ">=0.1.8,<0.2"
jaxtyping = ">=0.2.28,<0.3"
icecream = ">=2.1.3,<2.2"
eigen = ">=3.4.0,<3.5"
rerun-sdk = ">=0.16.1"
tyro = ">=0.8.4,<0.9"
unzip = ">=6.0,<7"
[pypi-dependencies]
opencv-python = ">=4.9.0.80"
evo = ">=1.28.0"
mini-dust3r = "*"
gradio-rerun = ">=0.0.3"
mmcv = "*"
yt-dlp = "*"
gradio = ">=4.36.0"
hf-transfer = "*"