[project] name = "mini-dpvo" version = "0.1.0" description = "Add a short description here" authors = ["pablovela5620 "] 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 = "*"