Spaces:
Starting
on
T4
Starting
on
T4
File size: 2,243 Bytes
2e3c4e2 a8c8616 2e3c4e2 9970917 2e3c4e2 899c526 a8c8616 899c526 a8c8616 899c526 2e45a07 a8c8616 0e99e24 2e3c4e2 a8c8616 2e3c4e2 9970917 2e3c4e2 9970917 a8c8616 2e3c4e2 9970917 a8c8616 24a3958 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 |
[project]
name = "mini-dpvo"
version = "0.1.0"
description = "Add a short description here"
authors = ["pablovela5620 <[email protected]>"]
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 = "*"
|