Spaces:
Running
on
Zero
Running
on
Zero
File size: 2,650 Bytes
fc9d64f cbc48ce da2f55c cbc48ce da2f55c cbc48ce da2f55c 4d80825 cbc48ce 2e7c76e fc9d64f 2e7c76e |
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 |
import os
import time
import subprocess
from pathlib import Path
REPO_ROOT = str(os.path.join(os.path.dirname(__file__)))
def run_cmds(cmds):
cmds = cmds.split('\n')
for cmd in cmds:
if len(cmd) == 0:
continue
os.system(cmd)
def prepare_env():
init_flag = Path(f'{REPO_ROOT}/initialized')
if init_flag.exists():
return
os.chdir(REPO_ROOT)
run_cmds(
f'''
git clone https://github.com/zju3dv/GVHMR --recursive {REPO_ROOT}/GVHMR
pip install -r {REPO_ROOT}/GVHMR/requirements.txt
pip install -e {REPO_ROOT}/GVHMR
mkdir {REPO_ROOT}/GVHMR/inputs
mkdir {REPO_ROOT}/GVHMR/outputs
ln -s {REPO_ROOT}/GVHMR/inputs {REPO_ROOT}/
ln -s {REPO_ROOT}/GVHMR/outputs {REPO_ROOT}/
ln -s {REPO_ROOT}/GVHMR/hmr4d {REPO_ROOT}/
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/SMPLer-X/resolve/main/SMPL_NEUTRAL.pkl -d {REPO_ROOT}/GVHMR/inputs/checkpoints/body_models/smpl -o SMPL_NEUTRAL.pkl
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/SMPLer-X/resolve/main/SMPLX_NEUTRAL.npz -d {REPO_ROOT}/GVHMR/inputs/checkpoints/body_models/smplx -o SMPLX_NEUTRAL.npz
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/GVHMR/resolve/main/dpvo/dpvo.pth -d {REPO_ROOT}/GVHMR/inputs/checkpoints/dpvo -o dpvo.pth
aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/GVHMR/resolve/main/gvhmr/gvhmr_siga24_release.ckpt -d {REPO_ROOT}/GVHMR/inputs/checkpoints/gvhmr -o gvhmr_siga24_release.ckpt
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
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
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
touch {REPO_ROOT}/initialized
'''
)
return
def server_up():
server_up_flag = Path(f'{REPO_ROOT}/server_up')
if server_up_flag.exists():
while True:
time.sleep(600)
else:
os.system(f'python {REPO_ROOT}/app/entry.py')
if __name__ == '__main__':
prepare_env()
server_up() |