Spaces:
Running
on
Zero
Running
on
Zero
IsshikiHugh
commited on
Commit
•
cbc48ce
1
Parent(s):
d607253
update
Browse files- app.py +12 -1
- app/gui.py +2 -2
- packages.txt +2 -1
app.py
CHANGED
@@ -1,4 +1,7 @@
|
|
1 |
import os
|
|
|
|
|
|
|
2 |
|
3 |
REPO_ROOT = str(os.path.join(os.path.dirname(__file__)))
|
4 |
|
@@ -11,6 +14,10 @@ def run_cmds(cmds):
|
|
11 |
|
12 |
|
13 |
def prepare_env():
|
|
|
|
|
|
|
|
|
14 |
os.chdir(REPO_ROOT)
|
15 |
run_cmds(
|
16 |
f'''
|
@@ -29,9 +36,13 @@ def prepare_env():
|
|
29 |
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/GVHMR/resolve/main/hmr2/epoch%3D10-step%3D25000.ckpt -d {REPO_ROOT}/GVHMR/inputs/checkpoints/hmr2 -o epoch=10-step=25000.ckpt
|
30 |
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/GVHMR/resolve/main/vitpose/vitpose-h-multi-coco.pth -d {REPO_ROOT}/GVHMR/inputs/checkpoints/vitpose -o vitpose-h-multi-coco.pth
|
31 |
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/GVHMR/resolve/main/yolo/yolov8x.pt -d {REPO_ROOT}/GVHMR/inputs/checkpoints/yolo -o yolov8x.pt
|
|
|
32 |
'''
|
33 |
)
|
34 |
|
|
|
|
|
35 |
if __name__ == '__main__':
|
36 |
prepare_env()
|
37 |
-
|
|
|
|
1 |
import os
|
2 |
+
import time
|
3 |
+
import subprocess
|
4 |
+
from pathlib import Path
|
5 |
|
6 |
REPO_ROOT = str(os.path.join(os.path.dirname(__file__)))
|
7 |
|
|
|
14 |
|
15 |
|
16 |
def prepare_env():
|
17 |
+
init_flag = Path(f'{REPO_ROOT}/initialized')
|
18 |
+
if init_flag.exists():
|
19 |
+
return
|
20 |
+
|
21 |
os.chdir(REPO_ROOT)
|
22 |
run_cmds(
|
23 |
f'''
|
|
|
36 |
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/GVHMR/resolve/main/hmr2/epoch%3D10-step%3D25000.ckpt -d {REPO_ROOT}/GVHMR/inputs/checkpoints/hmr2 -o epoch=10-step=25000.ckpt
|
37 |
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/GVHMR/resolve/main/vitpose/vitpose-h-multi-coco.pth -d {REPO_ROOT}/GVHMR/inputs/checkpoints/vitpose -o vitpose-h-multi-coco.pth
|
38 |
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/GVHMR/resolve/main/yolo/yolov8x.pt -d {REPO_ROOT}/GVHMR/inputs/checkpoints/yolo -o yolov8x.pt
|
39 |
+
touch {REPO_ROOT}/initialized
|
40 |
'''
|
41 |
)
|
42 |
|
43 |
+
return
|
44 |
+
|
45 |
if __name__ == '__main__':
|
46 |
prepare_env()
|
47 |
+
while True:
|
48 |
+
time.sleep(600)
|
app/gui.py
CHANGED
@@ -56,7 +56,7 @@ def get_desc():
|
|
56 |
<a href=https://arxiv.org/abs/2409.06662>Paper</a>
|
57 |
</b></center>
|
58 |
|
59 |
-
> World-Grounded Human Motion Recovery via Gravity-View Coordinates
|
60 |
> [Zehong Shen](https://zehongs.github.io/)<sup>\*</sup>,
|
61 |
[Huaijin Pi](https://phj128.github.io/)<sup>\*</sup>,
|
62 |
[Yan Xia](https://isshikihugh.github.io/scholar),
|
@@ -65,6 +65,6 @@ def get_desc():
|
|
65 |
[Zechen Hu](https://zju3dv.github.io/gvhmr),
|
66 |
[Hujun Bao](http://www.cad.zju.edu.cn/home/bao/),
|
67 |
[Ruizhen Hu](https://csse.szu.edu.cn/staff/ruizhenhu/),
|
68 |
-
[Xiaowei Zhou](https://xzhou.me/)
|
69 |
> SIGGRAPH Asia 2024
|
70 |
'''
|
|
|
56 |
<a href=https://arxiv.org/abs/2409.06662>Paper</a>
|
57 |
</b></center>
|
58 |
|
59 |
+
> World-Grounded Human Motion Recovery via Gravity-View Coordinates
|
60 |
> [Zehong Shen](https://zehongs.github.io/)<sup>\*</sup>,
|
61 |
[Huaijin Pi](https://phj128.github.io/)<sup>\*</sup>,
|
62 |
[Yan Xia](https://isshikihugh.github.io/scholar),
|
|
|
65 |
[Zechen Hu](https://zju3dv.github.io/gvhmr),
|
66 |
[Hujun Bao](http://www.cad.zju.edu.cn/home/bao/),
|
67 |
[Ruizhen Hu](https://csse.szu.edu.cn/staff/ruizhenhu/),
|
68 |
+
[Xiaowei Zhou](https://xzhou.me/)
|
69 |
> SIGGRAPH Asia 2024
|
70 |
'''
|
packages.txt
CHANGED
@@ -1,2 +1,3 @@
|
|
1 |
tmux
|
2 |
-
aria2
|
|
|
|
1 |
tmux
|
2 |
+
aria2
|
3 |
+
lsof
|