Spaces:
Runtime error
Runtime error
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
|