pablovela5620 commited on
Commit
9db2da3
·
verified ·
1 Parent(s): 650b0c0

Upload pyproject.toml with huggingface_hub

Browse files
Files changed (1) hide show
  1. pyproject.toml +135 -0
pyproject.toml ADDED
@@ -0,0 +1,135 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [build-system]
2
+ requires = ["setuptools>=61.0"]
3
+ build-backend = "setuptools.build_meta"
4
+
5
+ [project]
6
+ name = "mini-dpvo"
7
+ version = "0.1.1"
8
+ description = "Miniature version of dpvo, focused on inference"
9
+ requires-python = ">=3.10.0"
10
+ license = { file = "LICENSE" }
11
+ classifiers = [
12
+ "Development Status :: 3 - Alpha",
13
+ "Programming Language :: Python",
14
+ ]
15
+ readme = "README.md"
16
+ dependencies = ["build>=1.2.2.post1,<2", "mmcv-lite>=2.2.0,<3"]
17
+
18
+ [tool.pixi.project]
19
+ name = "mini-dpvo"
20
+ authors = ["pablovela5620 <pablovela5620@gmail.com>"]
21
+ channels = [
22
+ "nvidia/label/cuda-11.8.0",
23
+ "nvidia",
24
+ "conda-forge",
25
+ "pytorch",
26
+ "pyg",
27
+ ]
28
+ platforms = ["linux-64"]
29
+
30
+ [tool.pixi.system-requirements]
31
+ libc = { family = "glibc", version = "2.31" }
32
+
33
+ [tool.pixi.activation]
34
+ scripts = [".pixi.sh"]
35
+
36
+ [tool.pixi.tasks]
37
+ _download-model = """
38
+ test -e checkpoints/dpvo.pth
39
+ || (
40
+ wget https://www.dropbox.com/s/nap0u8zslspdwm4/models.zip
41
+ && unzip models.zip -d checkpoints
42
+ && rm -r models.zip
43
+ )
44
+ """
45
+ _download-dpvo-data = """
46
+ test -e data/movies/IMG_0492.MOV
47
+ || (
48
+ wget https://www.dropbox.com/s/7030y0mdl6efteg/movies.zip -P data/
49
+ && unzip data/movies.zip -d data/
50
+ && rm -r data/movies.zip
51
+ )
52
+ """
53
+
54
+ _download-iphone-data = """
55
+ test -e data/iphone/pool.MOV
56
+ || (
57
+ huggingface-cli download pablovela5620/dpvo-example-data pool.MOV --repo-type dataset --local-dir data/iphone/
58
+ )
59
+ """
60
+
61
+ post-install = { cmd = "pwd", depends_on = [
62
+ "_download-model",
63
+ "_download-dpvo-data",
64
+ "_download-iphone-data",
65
+ ] }
66
+
67
+ _build-wheel = "python -m build --no-isolation"
68
+ upload-hf = { cmd = "python tools/upload_to_hf.py", depends_on = [
69
+ "_build-wheel",
70
+ ] }
71
+
72
+ demo = """
73
+ python tools/demo.py --imagedir data/movies/IMG_0493.MOV --config config/fast.yaml
74
+ """
75
+ [tool.pixi.tasks.app]
76
+ cmd = "python tools/gradio_app.py"
77
+ description = "run gradio frontend for mini-dust3r"
78
+ depends_on = ["post-install"]
79
+
80
+ [tool.pixi.tasks.dev-app]
81
+ cmd = """
82
+ gradio tools/gradio_app.py
83
+ """
84
+ description = "Runs Gradio frontend with hot-reload"
85
+ depends_on = ["post-install"]
86
+
87
+
88
+ [tool.pixi.feature.spaces.tasks.app]
89
+ cmd = """
90
+ python gradio_app.py
91
+ """
92
+ description = "Runs Gradio frontend in huggingface spaces"
93
+ depends_on = ["post-install"]
94
+
95
+ [tool.pixi.pypi-options]
96
+ no-build-isolation = ["mini_dpvo"]
97
+
98
+ [tool.pixi.dependencies]
99
+ python = "3.11.*"
100
+ pip = ">=24.0,<25"
101
+ cuda = { version = "*", channel = "nvidia/label/cuda-11.8.0" }
102
+ pytorch-cuda = { version = "11.8.*", channel = "pytorch" }
103
+ pytorch = { version = ">=2.2.0,<2.3", channel = "pytorch" }
104
+ torchvision = { version = ">=0.17.0,<0.18", channel = "pytorch" }
105
+ plyfile = ">=1.0.3,<1.1"
106
+ pytorch-scatter = { version = "*", channel = "pyg" }
107
+ matplotlib = ">=3.8.4,<3.9"
108
+ yacs = ">=0.1.8,<0.2"
109
+ jaxtyping = ">=0.2.28,<0.3"
110
+ icecream = ">=2.1.3,<2.2"
111
+ eigen = ">=3.4.0,<3.5"
112
+ tyro = ">=0.8.4,<0.9"
113
+ numpy = "<2.0.0"
114
+ gradio = "<5"
115
+ rerun-sdk = ">=0.19.0,<0.20"
116
+
117
+ [tool.pixi.pypi-dependencies]
118
+ # mini_dpvo = { path = ".", editable = true }
119
+ mini_dpvo = { path = "dist/mini_dpvo-0.1.1-cp311-cp311-linux_x86_64.whl" }
120
+ opencv-python = ">=4.9.0.80"
121
+ evo = ">=1.28.0"
122
+ gradio-rerun = ">=0.0.3"
123
+ numpy = { version = "<2.0.0" }
124
+ mini-dust3r = ">=0.1.1, <0.2"
125
+
126
+ [tool.pixi.feature.spaces.pypi-dependencies]
127
+ spaces = "*"
128
+
129
+ [tool.pixi.environments]
130
+ spaces = { features = ["spaces"], solve-group = "default" }
131
+
132
+ [tool.ruff]
133
+ ignore = [
134
+ "F722", # Forward annotation false positive from jaxtyping. Should be caught by pyright.
135
+ ]