IsshikiHugh commited on
Commit
cbc48ce
1 Parent(s): d607253
Files changed (3) hide show
  1. app.py +12 -1
  2. app/gui.py +2 -2
  3. 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
- os.system('python app/entry.py')
 
 
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