floppyrj45/cosma-qc
2
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: dispatcher passe --save_poses à demo.py — génère lingbot_poses.npz (Plan 2)

Browse Source
This commit is contained in:
Flag
2026-04-24 08:07:55 +00:00
parent 4eb9f22813
commit 67643d05ff
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/dispatcher.py
View File

@@ -617,12 +617,13 @@ def do_reconstruct(job: sqlite3.Row, worker: dict, frames_dir: str) -> tuple[str
else: else:
window_size, overlap_size = 12, 3 window_size, overlap_size = 12, 3
marker = shlex.quote(frames_dir) marker = shlex.quote(frames_dir)
poses_path = f"{frames_dir}/lingbot_poses.npz"
cmd = ( cmd = (
f"cd {shlex.quote(worker['lingbot_path'])} && source .venv/bin/activate && " f"cd {shlex.quote(worker['lingbot_path'])} && source .venv/bin/activate && "
f"setsid python3 demo.py --model_path {shlex.quote(ckpt)} " f"setsid python3 demo.py --model_path {shlex.quote(ckpt)} "
f"--image_folder {shlex.quote(frames_dir)} --port {port} " f"--image_folder {shlex.quote(frames_dir)} --port {port} "
f"--stride {stride} --use_sdpa --mode windowed --window_size {window_size} --overlap_size {overlap_size} --offload_to_cpu " f"--stride {stride} --use_sdpa --mode windowed --window_size {window_size} --overlap_size {overlap_size} --offload_to_cpu "
f"--save_ply {shlex.quote(ply_path)} > {log} 2>&1 & " f"--save_ply {shlex.quote(ply_path)} --save_poses {shlex.quote(poses_path)} --save_poses_fps 2.0 > {log} 2>&1 & "
f"DEMO_PID=$!; " f"DEMO_PID=$!; "
f"for i in $(seq 1 3600); do " f"for i in $(seq 1 3600); do "
f" if ! kill -0 $DEMO_PID 2>/dev/null; then wait $DEMO_PID; exit $?; fi; " f" if ! kill -0 $DEMO_PID 2>/dev/null; then wait $DEMO_PID; exit $?; fi; "