cozyclay 1.9.0 → 2.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/CHANGELOG.md +171 -0
- package/bin/agent/agent-routes.mjs +74 -70
- package/bin/agent/agent-runner.mjs +10 -3
- package/bin/agent/motion-runtime.mjs +32 -260
- package/bin/agent/providers.mjs +46 -3
- package/bin/agent/session-store.mjs +41 -5
- package/bin/agent/studio-prompt.mjs +1 -1
- package/bin/agent/studio-tools.mjs +48 -11
- package/bin/cozyclay.mjs +26 -8
- package/bin/live/cli.mjs +22 -3
- package/bin/mcp-runtime.mjs +35 -20
- package/bin/telemetry-state.mjs +5 -2
- package/bin/update-check.mjs +15 -0
- package/dist/app/index.html +6 -6
- package/dist/assets/analytics-MVZOEW8O.js +1 -0
- package/dist/assets/app-C1sdVbg4.css +1 -0
- package/dist/assets/app-CA55tcI3.js +4876 -0
- package/dist/assets/{demo-B-eg1CT8.js → demo-ByeYp-dY.js} +1 -1
- package/dist/assets/{first-shot-handoff-CGgnvNu8.js → first-shot-handoff-DgT28ZiV.js} +1 -1
- package/dist/assets/{landing-IbRp3FnQ.js → landing-BaXkpFip.js} +1 -1
- package/dist/assets/shot-prompt-CbqWcWYi.css +1 -0
- package/dist/assets/shot-prompt-Yrt99wZx.js +17 -0
- package/dist/assets/{ticket-BgIzOBOf.js → ticket-BcmsnHn5.js} +1 -1
- package/dist/assets/workflow-DRdwoDBT.css +1 -0
- package/dist/assets/workflow-dNPGY9xz.js +179 -0
- package/dist/cozyclay-package.json +1 -1
- package/dist/fonts/IBM-Plex-Sans-OFL.txt +92 -0
- package/dist/fonts/JetBrains-Mono-OFL.txt +93 -0
- package/dist/fonts/README.md +13 -9
- package/dist/fonts/ibm-plex-sans-400-latin.woff2 +0 -0
- package/dist/fonts/ibm-plex-sans-500-latin.woff2 +0 -0
- package/dist/fonts/ibm-plex-sans-600-latin.woff2 +0 -0
- package/dist/fonts/jetbrains-mono-400-latin.woff2 +0 -0
- package/dist/fonts/jetbrains-mono-500-latin.woff2 +0 -0
- package/dist/index.html +7 -7
- package/dist/privacy/index.html +13 -8
- package/dist/sitemap.xml +6 -6
- package/dist/workflow/index.html +5 -5
- package/mcp/LIVE-PROTOCOL.md +1 -1
- package/mcp/live-hub.mjs +35 -93
- package/mcp/mesh-file.mjs +48 -0
- package/mcp/server.mjs +6 -77
- package/mcp/tool-handlers.mjs +224 -377
- package/package.json +2 -1
- package/src/App.jsx +1273 -9075
- package/src/analytics.js +393 -9
- package/src/app-context.js +211 -0
- package/src/app-stage.jsx +25 -24
- package/src/ardy/auto-fix-panel.css +108 -0
- package/src/ardy/collision-blockers.js +9 -3
- package/src/ardy/fix-collisions.js +10 -66
- package/src/ardy/ground.js +4 -4
- package/src/ardy/ik-drag.js +101 -0
- package/src/ardy/ik-key-json.js +43 -0
- package/src/ardy/ik.js +122 -76
- package/src/ardy/physics-panel.css +2 -0
- package/src/ardy/physics-panel.jsx +17 -5
- package/src/ardy/platform-fit-panel.css +64 -0
- package/src/ardy/platform-fit-panel.jsx +40 -0
- package/src/ardy/platform-fit.js +326 -0
- package/src/ardy/playback.js +166 -20
- package/src/ardy/range-pin.js +299 -0
- package/src/ardy/timeline-coordinates.js +26 -0
- package/src/ardy/timeline.css +1199 -0
- package/src/ardy/timeline.jsx +149 -41
- package/src/ardy/waypoints.js +2 -2
- package/src/asset-pane.css +462 -0
- package/src/asset-pane.jsx +461 -232
- package/src/command-bus.js +349 -0
- package/src/commands/ai.js +31 -0
- package/src/commands/cast.js +140 -0
- package/src/commands/elements/character.js +13 -0
- package/src/commands/elements/motion.js +14 -0
- package/src/commands/elements/object.js +22 -0
- package/src/commands/elements/scene.js +16 -0
- package/src/commands/elements/shot.js +9 -0
- package/src/commands/elements/stage.js +11 -0
- package/src/commands/elements.js +155 -0
- package/src/commands/export.js +20 -0
- package/src/commands/index.js +40 -0
- package/src/commands/motion.js +231 -0
- package/src/commands/objects.js +174 -0
- package/src/commands/project.js +53 -0
- package/src/commands/scene.js +81 -0
- package/src/commands/shared.js +19 -0
- package/src/commands/shot.js +103 -0
- package/src/commands/stage.js +17 -0
- package/src/commands/view.js +42 -0
- package/src/document-store.js +144 -0
- package/src/domains/cast.js +1007 -0
- package/src/domains/motion.js +3219 -0
- package/src/domains/objects.js +932 -0
- package/src/domains/scenes.js +817 -0
- package/src/domains/shots.js +404 -0
- package/src/domains/stage.js +93 -0
- package/src/dualview.jsx +113 -57
- package/src/facing-marks.js +3 -0
- package/src/first-success-guide.jsx +14 -14
- package/src/grid-view.js +18 -5
- package/src/hierarchy-model.js +24 -16
- package/src/hierarchy-panel.css +386 -0
- package/src/hierarchy-panel.jsx +150 -21
- package/src/{fal-motion-client.js → i2v-motion-client.js} +13 -13
- package/src/i2v-motion-studio.jsx +180 -0
- package/src/ik-camera.js +3 -0
- package/src/main.jsx +6 -0
- package/src/motion/generation.js +105 -0
- package/src/motion-readiness-ui.jsx +4 -4
- package/src/motion-readiness.js +11 -0
- package/src/motion-trail.js +366 -9
- package/src/object-gizmo.jsx +7 -1
- package/src/otio.js +2 -1
- package/src/panels/CameraPanel.jsx +60 -0
- package/src/panels/CharacterTransformPanel.jsx +48 -0
- package/src/panels/EnvironmentPanel.jsx +40 -0
- package/src/panels/Foldout.jsx +32 -0
- package/src/panels/LightPanel.jsx +24 -0
- package/src/panels/ObjectTransformPanel.jsx +554 -0
- package/src/panels/PosePanel.jsx +97 -0
- package/src/panels/ProjectPanel.jsx +33 -0
- package/src/panels/PromptBlocksPanel.jsx +442 -0
- package/src/panels/PropsPanel.jsx +70 -0
- package/src/panels/ReferenceImageField.jsx +91 -0
- package/src/panels/RigControlPanel.jsx +78 -0
- package/src/panels/RigPanel.jsx +37 -0
- package/src/panels/SubjectBox.jsx +47 -0
- package/src/panels/SubjectsPanel.jsx +39 -0
- package/src/panels/VideoCapturePanel.jsx +180 -0
- package/src/panels/details.css +1171 -0
- package/src/panels/motion.css +197 -0
- package/src/panels/pose.css +39 -0
- package/src/planview.jsx +60 -31
- package/src/posestudio.jsx +59 -4
- package/src/project-browser.css +1073 -0
- package/src/project-browser.jsx +284 -123
- package/src/range-pin-object-transform.js +19 -0
- package/src/range-pin-panel.css +518 -0
- package/src/range-pin-panel.jsx +344 -0
- package/src/result-modal.jsx +5 -5
- package/src/room.jsx +79 -51
- package/src/scene-objects.js +37 -1
- package/src/scenes.js +5 -0
- package/src/semantic-edit.js +2 -0
- package/src/settings-menu.jsx +39 -105
- package/src/shell/BottomDock.jsx +450 -0
- package/src/shell/DetailsSlot.jsx +513 -0
- package/src/shell/LibrarySlot.jsx +86 -0
- package/src/shell/MenuBar.jsx +601 -0
- package/src/shell/OutlinerSlot.jsx +50 -0
- package/src/shell/PreferencesDialog.jsx +538 -0
- package/src/shell/PreferencesSlot.jsx +40 -0
- package/src/shell/StatusBar.jsx +125 -0
- package/src/shell/StudioShell.jsx +92 -0
- package/src/shell/TopBar.jsx +129 -0
- package/src/shell/ViewportToolbar.jsx +656 -0
- package/src/shell/agent-glass.css +296 -0
- package/src/shell/dock.css +93 -0
- package/src/shell/glass-regions.css +985 -0
- package/src/shell/glass.css +687 -0
- package/src/shell/log-store.js +52 -0
- package/src/shell/mode.css +9 -0
- package/src/shell/preferences.css +651 -0
- package/src/shell/shell.css +282 -0
- package/src/shell/studio-shell-context.js +19 -0
- package/src/shell/topbar.css +435 -0
- package/src/shell/viewport.css +451 -0
- package/src/store/authored-intent.js +22 -0
- package/src/store/runtime-adapters.js +92 -0
- package/src/store/scene-stage.js +21 -0
- package/src/store/use-document-store.js +14 -0
- package/src/studio-actions.js +210 -0
- package/src/studio-agent-commands.js +41 -271
- package/src/studio-agent-context.js +39 -10
- package/src/studio-agent-motion.js +122 -373
- package/src/studio-agent-protocol.js +67 -31
- package/src/studio-app-binding.js +379 -0
- package/src/studio-contact-sheet.js +73 -0
- package/src/studio-elements.js +58 -51
- package/src/styles/themes.css +200 -0
- package/src/styles/tokens.css +69 -0
- package/src/styles.css +460 -1742
- package/src/theme.js +44 -0
- package/src/timeline-extent.js +16 -0
- package/src/trail-key-conflicts.js +39 -0
- package/src/trail-pick.js +59 -0
- package/src/ui.jsx +8 -8
- package/src/use-case-question.jsx +66 -0
- package/src/workflow/AgentPanel.jsx +82 -69
- package/src/workflow/agent-client.js +12 -2
- package/src/workflow/agent-panel.css +34 -32
- package/src/workflow/cozy-scene-node.css +2 -0
- package/src/workflow/workflow.css +2 -0
- package/tools/ardy/__pycache__/cclay_gvhmr_worker.cpython-313.pyc +0 -0
- package/tools/ardy/bridge.mjs +164 -146
- package/tools/ardy/visual-qa.mjs +5 -5
- package/tools/bench/EXP3.md +61 -0
- package/tools/bench/cclay_bench_extract_incam.py +125 -0
- package/tools/bench/cclay_bench_extract_obs.py +204 -0
- package/tools/bench/cclay_bench_runner.py +42 -0
- package/tools/bench/cube-contact.mjs +162 -0
- package/tools/bench/exp3.mjs +152 -0
- package/tools/bench/extract-bench-lib.mjs +216 -0
- package/tools/bench/extract-bench.mjs +302 -0
- package/tools/bench/fal-generate.mjs +88 -0
- package/tools/bench/fit/README.md +189 -0
- package/tools/bench/fit/camera.mjs +59 -0
- package/tools/bench/fit/contact.mjs +419 -0
- package/tools/bench/fit/footlock.mjs +234 -0
- package/tools/bench/fit/motion.mjs +85 -0
- package/tools/bench/fit/pin.mjs +27 -0
- package/tools/bench/fit/remote.mjs +105 -0
- package/tools/bench/fit-bench.mjs +126 -0
- package/tools/bench/fit-sanity.mjs +45 -0
- package/tools/bench/metrics.mjs +313 -0
- package/tools/bench/obs/depth.mjs +258 -0
- package/tools/bench/obs/extrinsics.mjs +175 -0
- package/tools/bench/obs/fit_mannequin_betas.py +340 -0
- package/tools/bench/obs/ground.mjs +294 -0
- package/tools/bench/obs/heading.mjs +151 -0
- package/tools/bench/obs/ladder.mjs +555 -0
- package/tools/bench/obs/mannequin-betas.json +124 -0
- package/tools/bench/obs/remote.mjs +83 -0
- package/tools/bench/obs/rest_joints.py +52 -0
- package/tools/bench/obs/ybot-targets.mjs +49 -0
- package/tools/bench/obs-bench.mjs +520 -0
- package/tools/bench/score.mjs +343 -0
- package/tools/bench/summarize.mjs +86 -0
- package/tools/dev/pages/privacy.html +13 -8
- package/tools/gt-render/browser.mjs +159 -0
- package/tools/gt-render/camera-math.mjs +276 -0
- package/tools/gt-render/page.mjs +267 -0
- package/tools/gt-render/render.mjs +514 -0
- package/tools/gt-render/scene-box.mjs +31 -0
- package/tools/gt-render/take-transform.mjs +87 -0
- package/tools/morphgs/assets/gen_truth.py +39 -0
- package/tools/morphgs/assets/mesh_ori_rig.txt +27 -0
- package/tools/morphgs/assets/playback-check.mjs +30 -0
- package/tools/morphgs/demo-gate.sh +26 -0
- package/tools/morphgs/fbx2morphgs.mjs +68 -0
- package/tools/morphgs/morphgs-to-cskel27.mjs +105 -0
- package/tools/morphgs/patches/preprocess_src-none-mode.patch +76 -0
- package/tools/morphgs/setup-on-cluster.sh +126 -0
- package/tools/qa/css-rule-usage.mjs +283 -0
- package/tools/qa/studio-control-count.mjs +70 -32
- package/tools/run-tests.mjs +134 -12
- package/tools/track/DESIGN.md +513 -0
- package/tools/track/backfill-provenance.mjs +116 -0
- package/tools/track/budget.mjs +8 -0
- package/tools/track/check-rig.mjs +209 -0
- package/tools/track/diagnostics.schema.json +60 -0
- package/tools/track/export-rig.mjs +256 -0
- package/tools/track/fallback.mjs +7 -0
- package/tools/track/fk-parity-fixture.mjs +161 -0
- package/tools/track/gate.mjs +349 -0
- package/tools/track/masks.mjs +318 -0
- package/tools/track/metrics.mjs +201 -0
- package/tools/track/publish-obs.mjs +108 -0
- package/tools/track/py/check_env.py +61 -0
- package/tools/track/py/eval_lr.py +229 -0
- package/tools/track/py/lr_viterbi.py +294 -0
- package/tools/track/py/masks.py +405 -0
- package/tools/track/py/objective.py +382 -0
- package/tools/track/py/rig.py +508 -0
- package/tools/track/py/scene.py +326 -0
- package/tools/track/py/test_joint_indices.py +123 -0
- package/tools/track/py/test_lr_viterbi.py +195 -0
- package/tools/track/py/test_masks.py +229 -0
- package/tools/track/py/test_rig.py +288 -0
- package/tools/track/py/test_scene.py +354 -0
- package/tools/track/py/test_track.py +328 -0
- package/tools/track/py/track.py +411 -0
- package/tools/track/remote.mjs +140 -0
- package/tools/track/rig-dump.mjs +227 -0
- package/tools/track/run-box-tests.mjs +14 -0
- package/tools/track/run-box.mjs +168 -0
- package/tools/track/setup-box.sh +59 -0
- package/tools/track/study-2d.mjs +728 -0
- package/dist/assets/analytics-B1hnH66c.js +0 -1
- package/dist/assets/app-BWusbqgO.js +0 -4861
- package/dist/assets/app-qKDo4PBX.css +0 -1
- package/dist/assets/shot-prompt-CHUtw6af.js +0 -17
- package/dist/assets/shot-prompt-C_g2BHVc.css +0 -1
- package/dist/assets/workflow-CVwzMhz2.js +0 -179
- package/dist/assets/workflow-DsudxKHF.css +0 -1
- package/src/fal-motion-studio.jsx +0 -180
- package/src/scene-history.js +0 -129
|
@@ -0,0 +1,520 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
/**
|
|
3
|
+
* obs-bench.mjs — observation-ladder bench (#462). For each approved item:
|
|
4
|
+
* extract the obs NPZ on the GPU box (serial, tools/bench/obs/remote.mjs),
|
|
5
|
+
* fetch/cache the SMPL rest joints of its betas (tools/bench/obs/rest_joints.py),
|
|
6
|
+
* build every requested ladder step (tools/bench/obs/ladder.mjs), write the
|
|
7
|
+
* cskel27 take and score it.
|
|
8
|
+
*
|
|
9
|
+
* <out>/<set>/<name>/
|
|
10
|
+
* bench.log timestamped item log
|
|
11
|
+
* obs-base/{obs.npz,manifest.json,extract.log} GVHMR's own betas
|
|
12
|
+
* obs-mannequin/{obs.npz,betas.json,manifest.json,extract.log}
|
|
13
|
+
* input/video.mp4 fal only: normalised to the camera size, 24 fps
|
|
14
|
+
* <step>/{motion.npz,result.json,score.log,score/score.json}
|
|
15
|
+
*
|
|
16
|
+
* Only user-known inputs reach a step: the video, camera.json, the character
|
|
17
|
+
* body (canonical cskel27 rest skeleton, mannequin betas), scene.json boxes
|
|
18
|
+
* and the FIRST and LAST poses of the A/B source npz. Ground-truth joints,
|
|
19
|
+
* masks and intermediate frames are read by the scorers only.
|
|
20
|
+
*/
|
|
21
|
+
import { spawnSync } from "node:child_process";
|
|
22
|
+
import { createHash } from "node:crypto";
|
|
23
|
+
import { appendFileSync, closeSync, copyFileSync, existsSync, mkdirSync, openSync, readFileSync, writeFileSync, rmSync } from "node:fs";
|
|
24
|
+
import { basename, dirname, join, resolve } from "node:path";
|
|
25
|
+
import { setTimeout as sleep } from "node:timers/promises";
|
|
26
|
+
import { fileURLToPath } from "node:url";
|
|
27
|
+
import { motionArraysToNpzMembers, writeNpz } from "../ardy/npz.mjs";
|
|
28
|
+
import { readNpz } from "../kimodo/read-npz.mjs";
|
|
29
|
+
import { backgroundDifferenceMask, paletteMask, PALETTE_THRESHOLDS } from "./cube-contact.mjs";
|
|
30
|
+
import * as THREE from "three";
|
|
31
|
+
import { FBXLoader } from "three/examples/jsm/loaders/FBXLoader.js";
|
|
32
|
+
import { applyMotionFrame } from "../../src/ardy/playback.js";
|
|
33
|
+
import { readEndpoints, regenerateJoints } from "./fit/motion.mjs";
|
|
34
|
+
import { maskIoU, mean, meanJointError } from "./metrics.mjs";
|
|
35
|
+
import { ladderStep, restInfo, STEPS } from "./obs/ladder.mjs";
|
|
36
|
+
import { extractObs } from "./obs/remote.mjs";
|
|
37
|
+
import { ensureMasks } from "../../tools/track/masks.mjs";
|
|
38
|
+
import { fallbackStep } from "../../tools/track/fallback.mjs";
|
|
39
|
+
import { runTracker, TRACK_ABLATIONS } from "../../tools/track/remote.mjs";
|
|
40
|
+
|
|
41
|
+
const HERE = dirname(fileURLToPath(import.meta.url));
|
|
42
|
+
const ROOT = resolve(HERE, "../..");
|
|
43
|
+
const WRAPPER = join(HERE, "cclay_bench_extract_obs.py");
|
|
44
|
+
const REST_SCRIPT = join(HERE, "obs/rest_joints.py");
|
|
45
|
+
const MANNEQUIN = join(HERE, "obs/mannequin-betas.json");
|
|
46
|
+
const SSH = ["-o", "BatchMode=yes", "-o", "ConnectTimeout=15", ...(process.env.CCLAY_EXTRACT_SSH_PORT ? ["-p", process.env.CCLAY_EXTRACT_SSH_PORT] : [])];
|
|
47
|
+
const MANNEQUIN_STEPS = new Set(["G5", "Gbest", "T1"]);
|
|
48
|
+
export const OBS_BENCH_STEPS = Object.freeze([...STEPS, "T1"]);
|
|
49
|
+
export const DEFAULT_OBS_ROOT = process.env.COZYCLAY_OBS_ROOT ?? "evidence/obs/cache";
|
|
50
|
+
|
|
51
|
+
export const USAGE = `usage: node tools/bench/obs-bench.mjs --approved <approved.json> --out <dir> [options]
|
|
52
|
+
|
|
53
|
+
--items a,b item names (or set/name); default: every approved item
|
|
54
|
+
--steps G0,... any of ${OBS_BENCH_STEPS.join(",")} (default: all)
|
|
55
|
+
--host <ssh-dest> GPU box (default yun@ubuntu-baremetal)
|
|
56
|
+
--rest-cache <dir> SMPL rest-joint cache (default: <out>/../rest)
|
|
57
|
+
--port 5198 Vite port for the scorers
|
|
58
|
+
--cdp-port 9238 headless Chrome CDP port for the scorers
|
|
59
|
+
--obs-root <dir> read obs from <dir>/<set>/<name>/{base,g5}/obs.npz, the serial sweep's cache
|
|
60
|
+
(default ${DEFAULT_OBS_ROOT}); a missing file fails the item. Never touches the GPU.
|
|
61
|
+
--extract instead of --obs-root, run GVHMR on the box (serial; waits for a busy GPU)
|
|
62
|
+
--gpu-wait-min 60 with --extract: wait this long for a busy GPU before failing the item
|
|
63
|
+
--force-obs with --extract: re-extract even when the manifest matches
|
|
64
|
+
--character <model> character rig (public/models/<model>.fbx) whose standing ankle height is the G3 floor plane (default y-bot-tpose)
|
|
65
|
+
--track-ablate full T1 fit components, one of ${TRACK_ABLATIONS.join(",")} (cumulative; default full)
|
|
66
|
+
--force rebuild and rescore steps whose result.json says ok
|
|
67
|
+
--help this text
|
|
68
|
+
|
|
69
|
+
Idempotent: a step whose result.json has ok=true is skipped; obs NPZs are
|
|
70
|
+
reused while video, K, betas and the wrapper are unchanged. One GPU
|
|
71
|
+
extraction at a time; items run serially.`;
|
|
72
|
+
|
|
73
|
+
export function parseArgs(argv) {
|
|
74
|
+
const options = { steps: [...OBS_BENCH_STEPS], host: "yun@ubuntu-baremetal", port: 5198, cdpPort: 9238, gpuWaitMin: 60, force: false, forceObs: false, obsRoot: DEFAULT_OBS_ROOT, extract: false, character: "y-bot-tpose", trackAblate: "full" };
|
|
75
|
+
const names = { "--approved": "approved", "--out": "out", "--items": "items", "--steps": "steps", "--host": "host", "--rest-cache": "restCache", "--port": "port", "--cdp-port": "cdpPort", "--gpu-wait-min": "gpuWaitMin", "--obs-root": "obsRoot", "--character": "character", "--track-ablate": "trackAblate" };
|
|
76
|
+
for (let i = 0; i < argv.length; i++) {
|
|
77
|
+
const flag = argv[i];
|
|
78
|
+
if (flag === "--help" || flag === "-h") { options.help = true; continue; }
|
|
79
|
+
if (flag === "--force") { options.force = true; continue; }
|
|
80
|
+
if (flag === "--force-obs") { options.forceObs = true; continue; }
|
|
81
|
+
if (flag === "--extract") { options.extract = true; continue; }
|
|
82
|
+
if (!names[flag]) throw new Error(`unknown option ${flag}`);
|
|
83
|
+
const value = argv[++i];
|
|
84
|
+
if (value === undefined || value.startsWith("--")) throw new Error(`${flag} needs a value`);
|
|
85
|
+
options[names[flag]] = value;
|
|
86
|
+
}
|
|
87
|
+
if (options.help) return options;
|
|
88
|
+
for (const key of ["approved", "out"]) if (!options[key]) throw new Error(`--${key} is required`);
|
|
89
|
+
if (typeof options.steps === "string") options.steps = options.steps.split(",").map((s) => s.trim()).filter(Boolean);
|
|
90
|
+
const unknown = options.steps.filter((s) => !OBS_BENCH_STEPS.includes(s));
|
|
91
|
+
if (unknown.length || !options.steps.length) throw new Error(`unknown steps ${unknown.join(",")}; known: ${OBS_BENCH_STEPS.join(",")}`);
|
|
92
|
+
options.steps = OBS_BENCH_STEPS.filter((s) => options.steps.includes(s));
|
|
93
|
+
if (!TRACK_ABLATIONS.includes(options.trackAblate)) throw new Error(`--track-ablate must be one of ${TRACK_ABLATIONS.join(",")}`);
|
|
94
|
+
if (typeof options.items === "string") options.items = options.items.split(",").map((s) => s.trim()).filter(Boolean);
|
|
95
|
+
for (const key of ["port", "cdpPort", "gpuWaitMin"]) {
|
|
96
|
+
options[key] = Number(options[key]);
|
|
97
|
+
if (!(Number.isFinite(options[key]) && options[key] >= 0)) throw new Error(`--${key} must be a nonnegative number`);
|
|
98
|
+
}
|
|
99
|
+
options.approved = resolve(options.approved);
|
|
100
|
+
options.out = resolve(options.out);
|
|
101
|
+
options.restCache = resolve(options.restCache ?? join(dirname(options.out), "rest"));
|
|
102
|
+
if (options.extract && argv.includes("--obs-root")) throw new Error("--extract and --obs-root are exclusive");
|
|
103
|
+
options.obsRoot = options.extract ? null : resolve(options.obsRoot);
|
|
104
|
+
return options;
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
export function selectItems(approved, names) {
|
|
108
|
+
const items = approved.items;
|
|
109
|
+
if (!names) return items;
|
|
110
|
+
return names.map((name) => {
|
|
111
|
+
const found = items.filter((item) => name === item.name || name === `${item.set}/${item.name}`);
|
|
112
|
+
if (!found.length) throw new Error(`no approved item ${name}`);
|
|
113
|
+
if (found.length > 1) throw new Error(`ambiguous item ${name}: use one of ${found.map((i) => `${i.set}/${i.name}`).join(", ")}`);
|
|
114
|
+
return found[0];
|
|
115
|
+
});
|
|
116
|
+
}
|
|
117
|
+
|
|
118
|
+
const readJson = (path) => (existsSync(path) ? JSON.parse(readFileSync(path, "utf8")) : null);
|
|
119
|
+
const putJson = (path, value) => { mkdirSync(dirname(path), { recursive: true }); writeFileSync(path, `${JSON.stringify(value, null, "\t")}\n`); };
|
|
120
|
+
const sha256 = (...parts) => { const h = createHash("sha256"); for (const p of parts) h.update(p); return h.digest("hex"); };
|
|
121
|
+
const fileSha = (path) => sha256(readFileSync(path));
|
|
122
|
+
|
|
123
|
+
function commitId() {
|
|
124
|
+
const head = spawnSync("git", ["rev-parse", "HEAD"], { cwd: ROOT, encoding: "utf8" }).stdout.trim();
|
|
125
|
+
const dirty = spawnSync("git", ["status", "--porcelain"], { cwd: ROOT, encoding: "utf8" }).stdout.trim();
|
|
126
|
+
return head ? `${head}${dirty ? "-dirty" : ""}` : null;
|
|
127
|
+
}
|
|
128
|
+
|
|
129
|
+
/** Everything a real user has for this item, as paths. */
|
|
130
|
+
export function itemInputs(item, itemDir) {
|
|
131
|
+
const camera = join(item.dir, item.variant, "camera.json");
|
|
132
|
+
// fal items carry no source npz; their A/B poses are the scenario's GT npz (as exp3 used them).
|
|
133
|
+
const source = item.source ?? join(dirname(dirname(item.dir)), "gt-motions", `${basename(item.dir)}.npz`);
|
|
134
|
+
return {
|
|
135
|
+
set: item.set,
|
|
136
|
+
name: item.name,
|
|
137
|
+
camera,
|
|
138
|
+
video: item.set === "fal" ? join(itemDir, "input", "video.mp4") : join(item.dir, item.variant, "video.mp4"),
|
|
139
|
+
rawVideo: item.video ?? join(item.dir, item.variant, "video.mp4"),
|
|
140
|
+
source,
|
|
141
|
+
scene: item.scene ?? null,
|
|
142
|
+
// Skin renders are invisible to the palette detector (exp3 base condition yolo-vitpose).
|
|
143
|
+
detector: item.variant === "skin" ? "yolo" : "palette",
|
|
144
|
+
};
|
|
145
|
+
}
|
|
146
|
+
|
|
147
|
+
function ffprobe(path) {
|
|
148
|
+
const out = spawnSync("ffprobe", ["-v", "error", "-select_streams", "v:0", "-show_entries", "stream=width,height,nb_frames", "-of", "json", path], { encoding: "utf8" });
|
|
149
|
+
if (out.status) throw new Error(`ffprobe ${path}: ${out.stderr}`);
|
|
150
|
+
return JSON.parse(out.stdout).streams[0];
|
|
151
|
+
}
|
|
152
|
+
|
|
153
|
+
/** fal clips come at the service's size/rate: bring them to the known camera's image and 24 fps (exp3's normalisation). */
|
|
154
|
+
function normalizeFalVideo(inputs, log) {
|
|
155
|
+
if (existsSync(inputs.video)) return;
|
|
156
|
+
const camera = readJson(inputs.camera), probe = ffprobe(inputs.rawVideo);
|
|
157
|
+
if (Math.abs(probe.width / probe.height / (camera.width / camera.height) - 1) > 0.01) throw new Error("fal changed the aspect ratio; the known camera does not apply");
|
|
158
|
+
mkdirSync(dirname(inputs.video), { recursive: true });
|
|
159
|
+
const run = spawnSync("ffmpeg", ["-v", "error", "-y", "-i", inputs.rawVideo, "-vf", `scale=${camera.width}:${camera.height},fps=24`, "-c:v", "libx264", "-crf", "12", "-pix_fmt", "yuv420p", inputs.video], { encoding: "utf8" });
|
|
160
|
+
if (run.status) throw new Error(`ffmpeg normalise failed: ${run.stderr}`);
|
|
161
|
+
log(`normalised ${inputs.rawVideo} -> ${inputs.video}`);
|
|
162
|
+
}
|
|
163
|
+
|
|
164
|
+
async function ensureObs({ kind, inputs, itemDir, options, log }) {
|
|
165
|
+
const dir = join(itemDir, `obs-${kind}`), output = join(dir, "obs.npz"), manifestPath = join(dir, "manifest.json");
|
|
166
|
+
mkdirSync(dir, { recursive: true });
|
|
167
|
+
let betasPath = null, betas = null;
|
|
168
|
+
if (kind === "mannequin") {
|
|
169
|
+
betas = readJson(MANNEQUIN).betas;
|
|
170
|
+
betasPath = join(dir, "betas.json");
|
|
171
|
+
writeFileSync(betasPath, `${JSON.stringify(betas)}\n`); // the wrapper wants a bare 10-float list
|
|
172
|
+
}
|
|
173
|
+
const K = readJson(inputs.camera).K;
|
|
174
|
+
const signature = sha256(fileSha(inputs.video), JSON.stringify(K), JSON.stringify(betas), fileSha(WRAPPER), inputs.detector, "vitpose");
|
|
175
|
+
const manifest = readJson(manifestPath);
|
|
176
|
+
// The shared obs root (the control session's serial sweep cache) is the
|
|
177
|
+
// default and is authoritative: read it, never contend for the GPU. It is
|
|
178
|
+
// checked first so a re-extracted cache file replaces a stale local copy.
|
|
179
|
+
if (options.obsRoot) {
|
|
180
|
+
const shared = join(options.obsRoot, inputs.set, inputs.name, kind === "mannequin" ? "g5" : "base", "obs.npz");
|
|
181
|
+
if (!existsSync(shared)) throw new Error(`obs-${kind}: missing ${shared}; the obs sweep has not produced it (see ${join(dirname(shared), "extract.log")}). This bench never extracts when --obs-root is in use; pass --extract to run GVHMR itself.`);
|
|
182
|
+
const sourceSha256 = fileSha(shared);
|
|
183
|
+
if (betas) {
|
|
184
|
+
const used = Array.from(readNpz(shared).betas_used.data);
|
|
185
|
+
const gap = Math.max(...used.map((v, i) => Math.abs(v - betas[i])));
|
|
186
|
+
if (!(gap < 1e-5)) throw new Error(`obs-${kind}: ${shared} was extracted with betas ${JSON.stringify(used)}, not the current ${MANNEQUIN} (max diff ${gap}); re-run the sweep for this item`);
|
|
187
|
+
}
|
|
188
|
+
if (existsSync(output) && manifest?.sourceSha256 === sourceSha256) {
|
|
189
|
+
log(`obs-${kind}: cached (${output} = ${shared})`);
|
|
190
|
+
return { path: output, manifest };
|
|
191
|
+
}
|
|
192
|
+
const record = { signature, source: shared, sourceSha256, video: inputs.video, camera: inputs.camera, K, betas, detector: inputs.detector, keypoints: "vitpose", createdAt: new Date().toISOString() };
|
|
193
|
+
copyFileSync(shared, output);
|
|
194
|
+
putJson(manifestPath, record);
|
|
195
|
+
log(`obs-${kind}: from obs root (${shared})`);
|
|
196
|
+
return { path: output, manifest: record };
|
|
197
|
+
}
|
|
198
|
+
if (!options.forceObs && existsSync(output) && manifest?.signature === signature && !manifest.source) {
|
|
199
|
+
log(`obs-${kind}: cached (${output})`);
|
|
200
|
+
return { path: output, manifest };
|
|
201
|
+
}
|
|
202
|
+
const deadline = Date.now() + options.gpuWaitMin * 60000;
|
|
203
|
+
for (;;) {
|
|
204
|
+
try {
|
|
205
|
+
log(`obs-${kind}: extracting on ${options.host} (${inputs.detector}+vitpose${betas ? ", mannequin betas" : ""})`);
|
|
206
|
+
const started = Date.now();
|
|
207
|
+
const run = await extractObs({ host: options.host, video: inputs.video, camera: inputs.camera, output, detector: inputs.detector, keypoints: "vitpose", betas: betasPath, log: join(dir, "extract.log") });
|
|
208
|
+
const record = { signature, video: inputs.video, camera: inputs.camera, K, betas, detector: inputs.detector, keypoints: "vitpose", wrapperSha256: fileSha(WRAPPER), remote: run.remote, seconds: (Date.now() - started) / 1000, createdAt: new Date().toISOString() };
|
|
209
|
+
putJson(manifestPath, record);
|
|
210
|
+
log(`obs-${kind}: done in ${record.seconds.toFixed(1)} s`);
|
|
211
|
+
return { path: output, manifest: record };
|
|
212
|
+
} catch (error) {
|
|
213
|
+
// Another sweep holds the GPU (refused up front, or it started between our
|
|
214
|
+
// idle check and our allocation): wait, never kill it.
|
|
215
|
+
const busy = /GPU busy|CUDA out of memory/;
|
|
216
|
+
if (!busy.test(String(error.message)) || Date.now() > deadline) throw error;
|
|
217
|
+
log(`obs-${kind}: GPU busy; retrying in 20 s (${String(error.message).split("\n").find((l) => busy.test(l))?.trim().slice(0, 300)})`);
|
|
218
|
+
await sleep(20000);
|
|
219
|
+
}
|
|
220
|
+
}
|
|
221
|
+
}
|
|
222
|
+
|
|
223
|
+
const rigs = new Map();
|
|
224
|
+
/** The character's standing ankle-bone height: the A and B poses (user-known)
|
|
225
|
+
* played on the character rig with the Studio's own playback code, lower
|
|
226
|
+
* foot bone of each, averaged. This is where the rendered character's ankle
|
|
227
|
+
* actually is (walk: 0.141 / 0.131 m, the render measures 0.141 at frame 0);
|
|
228
|
+
* the canonical cskel27 skeleton's 0.058 m is not the rendered body. */
|
|
229
|
+
export function characterStandingAnkle(endpoints, model = "y-bot-tpose") {
|
|
230
|
+
if (!rigs.has(model)) {
|
|
231
|
+
const path = ["public", "dist"].map((d) => join(ROOT, d, "models", `${model}.fbx`)).find(existsSync);
|
|
232
|
+
if (!path) throw new Error(`character model ${model}.fbx not found under public/ or dist/models`);
|
|
233
|
+
const bytes = readFileSync(path);
|
|
234
|
+
const rig = new FBXLoader().parse(bytes.buffer.slice(bytes.byteOffset, bytes.byteOffset + bytes.byteLength), "");
|
|
235
|
+
rig.scale.setScalar(0.01);
|
|
236
|
+
rig.updateMatrixWorld(true);
|
|
237
|
+
rigs.set(model, { rig, path });
|
|
238
|
+
}
|
|
239
|
+
const { rig, path } = rigs.get(model);
|
|
240
|
+
const bone = (name) => { let found = null; rig.traverse((o) => { if (o.isBone && o.name === name) found = o; }); if (!found) throw new Error(`${model}: no bone ${name}`); return found; };
|
|
241
|
+
const feet = [bone("mixamorigLeftFoot"), bone("mixamorigRightFoot")], position = new THREE.Vector3();
|
|
242
|
+
const perPose = endpoints.map((pose) => {
|
|
243
|
+
const motion = regenerateJoints({ frames: 1, fps: 24, rotMats: Float32Array.from(pose.rotMats), rootPos: Float32Array.from(pose.rootPos), posedJoints: new Float32Array(81), boneScale: new Float32Array(27).fill(1) });
|
|
244
|
+
applyMotionFrame(rig, motion, 0);
|
|
245
|
+
rig.updateMatrixWorld(true);
|
|
246
|
+
return Math.min(...feet.map((b) => position.setFromMatrixPosition(b.matrixWorld).y));
|
|
247
|
+
});
|
|
248
|
+
return { height: mean(perPose), perPose, model: path, method: "lower mixamo foot bone of A and B through src/ardy/playback.js applyMotionFrame" };
|
|
249
|
+
}
|
|
250
|
+
|
|
251
|
+
/** SMPL rest joints for `betas`, computed on the box once and cached by (betas, helper) hash. */
|
|
252
|
+
function ensureRest(betas, options, log) {
|
|
253
|
+
const key = sha256(Buffer.from(Float32Array.from(betas).buffer), readFileSync(REST_SCRIPT));
|
|
254
|
+
const path = join(options.restCache, `${key}.json`);
|
|
255
|
+
const cached = readJson(path);
|
|
256
|
+
if (cached) return { rest: restInfo(cached), path, key };
|
|
257
|
+
const list = JSON.stringify(Array.from(Float32Array.from(betas)));
|
|
258
|
+
const run = spawnSync("ssh", [...SSH, options.host, `cd ~/cclay-ingest/GVHMR && CUDA_VISIBLE_DEVICES= .venv/bin/python - '${list}'`], { input: readFileSync(REST_SCRIPT), encoding: "utf8", timeout: 300000 });
|
|
259
|
+
if (run.status !== 0) throw new Error(`rest_joints.py failed (${run.status}): ${run.stderr?.slice(-2000)}`);
|
|
260
|
+
const value = JSON.parse(run.stdout.trim().split("\n").at(-1));
|
|
261
|
+
putJson(path, { ...value, key, createdAt: new Date().toISOString() });
|
|
262
|
+
log(`rest joints: computed ${path}`);
|
|
263
|
+
return { rest: restInfo(value), path, key };
|
|
264
|
+
}
|
|
265
|
+
|
|
266
|
+
function sceneBoxes(scenePath) {
|
|
267
|
+
if (!scenePath) return { boxes: [], placement: null };
|
|
268
|
+
const scene = readJson(scenePath);
|
|
269
|
+
return { boxes: [{ min: scene.min, max: scene.max }], placement: scene.placement };
|
|
270
|
+
}
|
|
271
|
+
|
|
272
|
+
function runLogged(args, logPath) {
|
|
273
|
+
const fd = openSync(logPath, "a");
|
|
274
|
+
try {
|
|
275
|
+
const run = spawnSync(process.execPath, args, { cwd: ROOT, stdio: ["ignore", fd, fd], env: { ...process.env, COZYCLAY_LIVE_PORT: process.env.COZYCLAY_LIVE_PORT || String(5300 + 462) }, timeout: 30 * 60000 });
|
|
276
|
+
if (run.status !== 0) throw new Error(`${basename(args[0])} exited ${run.status ?? run.signal}; see ${logPath}`);
|
|
277
|
+
} finally { closeSync(fd); }
|
|
278
|
+
}
|
|
279
|
+
|
|
280
|
+
/** gt / cube: tools/bench/score.mjs against the rendered GT (with the scene box for cube). */
|
|
281
|
+
function scoreAgainstGt({ item, inputs, motionPath, stepDir, options }) {
|
|
282
|
+
const out = join(stepDir, "score");
|
|
283
|
+
const args = [join(HERE, "score.mjs"), "--gt", join(item.dir, item.variant), "--pred", motionPath, "--out", out, "--port", String(options.port), "--cdp-port", String(options.cdpPort)];
|
|
284
|
+
if (inputs.scene) args.push("--box", JSON.stringify(sceneBoxes(inputs.scene).boxes[0]), "--gt-npz", inputs.source);
|
|
285
|
+
runLogged(args, join(stepDir, "score.log"));
|
|
286
|
+
const s = readJson(join(out, "score.json"));
|
|
287
|
+
return {
|
|
288
|
+
ateAlignedRmseM: s.trajectory.ateAlignedM.rmse, rootErrorRawRmseM: s.trajectory.rootErrorRawM.rmse, pathLengthRatio: s.trajectory.pathLengthRatio, displacementRatio: s.trajectory.displacementRatio,
|
|
289
|
+
paMpjpeM: s.pose.paMpjpeM, mpjpeAlignedM: s.pose.mpjpeRootRelativeAlignedM, iouRaw: s.overlap.maskIoURawMean, iouAligned: s.overlap.maskIoUAlignedMean,
|
|
290
|
+
endpointFirstRawM: s.endpoints.firstFrame.jointErrorRawM, endpointLastRawM: s.endpoints.lastFrame.jointErrorRawM, endpointLastAlignedM: s.endpoints.lastFrame.jointErrorAlignedM,
|
|
291
|
+
contact: s.contact ? { predRaw: s.contact.predRaw, gt: s.contact.gt } : null,
|
|
292
|
+
};
|
|
293
|
+
}
|
|
294
|
+
|
|
295
|
+
/** fal: exp3's fal scoring (no motion GT): A/B endpoint joint error, whole-clip box contact, unaligned proxy-mask IoU. */
|
|
296
|
+
export function scoreFal({ item, inputs, motionPath, stepDir, options }) {
|
|
297
|
+
const out = join(stepDir, "score"), camera = readJson(inputs.camera), { boxes, placement } = sceneBoxes(inputs.scene);
|
|
298
|
+
const args = [join(ROOT, "tools/gt-render/render.mjs"), "--out", out, "--no-video", "--camera", inputs.camera, "--port", String(options.port), "--cdp-port", String(options.cdpPort)];
|
|
299
|
+
if (placement) args.push("--scene-box", JSON.stringify(placement), "--box", JSON.stringify(boxes[0]));
|
|
300
|
+
args.push(motionPath);
|
|
301
|
+
runLogged(args, join(stepDir, "score.log"));
|
|
302
|
+
const rendered = join(out, basename(motionPath, ".npz"));
|
|
303
|
+
const joints = readJson(join(rendered, "joints.json")), contact = readJson(join(rendered, "contact.json"));
|
|
304
|
+
const ff = (a) => spawnSync("ffmpeg", ["-v", "error", ...a], { maxBuffer: 1 << 30 }).stdout;
|
|
305
|
+
const size = camera.width * camera.height, frames = Number(ffprobe(inputs.video).nb_frames);
|
|
306
|
+
const rgb = ff(["-i", inputs.video, "-f", "rawvideo", "-pix_fmt", "rgb24", "pipe:1"]);
|
|
307
|
+
const plate = item.variant === "skin" ? ff(["-i", join(item.dir, "plate.png"), "-f", "rawvideo", "-pix_fmt", "rgb24", "pipe:1"]) : null;
|
|
308
|
+
const masks = plate ? backgroundDifferenceMask(rgb, plate) : paletteMask(rgb);
|
|
309
|
+
const predMask = ff(["-i", join(rendered, "mask/%06d.png"), "-f", "rawvideo", "-pix_fmt", "gray", "pipe:1"]);
|
|
310
|
+
const count = Math.min(frames, joints.frames);
|
|
311
|
+
const ious = Array.from({ length: count }, (_, i) => maskIoU(masks.subarray(i * size, (i + 1) * size), predMask.subarray(i * size, (i + 1) * size)).iou);
|
|
312
|
+
// A/B as rendered: the GT render's first and last frames only.
|
|
313
|
+
const endpoints = readJson(join(item.dir, item.variant, "joints.json"));
|
|
314
|
+
const score = {
|
|
315
|
+
clip: item.name, variant: item.variant, noMotionGroundTruth: true,
|
|
316
|
+
endpointFirstM: meanJointError(joints.world[0], endpoints.world[0]), endpointLastM: meanJointError(joints.world.at(-1), endpoints.world.at(-1)),
|
|
317
|
+
minDistanceM: contact ? Math.min(...contact.frames.map((v) => v.minDistanceM)) : null, maxPenetrationM: contact ? Math.max(...contact.frames.map((v) => v.maxPenetrationM)) : null,
|
|
318
|
+
contactWindow: "entire clip: fal contact timing is unknown", overlapIoU: mean(ious),
|
|
319
|
+
segmentation: plate ? { method: "background-difference", minChannelDifference: 30, plate: join(item.dir, "plate.png") } : { method: "palette-colour", ...PALETTE_THRESHOLDS },
|
|
320
|
+
comparedFrames: count, sourceFrames: frames, predictionFrames: joints.frames, alignment: "none; fixed A-still camera", cameraAssumption: "no H3 camera drift", scorer: "tools/bench/obs-bench.mjs scoreFal (exp3 fal scoring)",
|
|
321
|
+
};
|
|
322
|
+
putJson(join(out, "score.json"), score);
|
|
323
|
+
return { endpointFirstM: score.endpointFirstM, endpointLastM: score.endpointLastM, minDistanceM: score.minDistanceM, maxPenetrationM: score.maxPenetrationM, iou: score.overlapIoU };
|
|
324
|
+
}
|
|
325
|
+
|
|
326
|
+
export function trackerRigPath(character = "y-bot-tpose") {
|
|
327
|
+
return join(ROOT, "node_modules", ".cache", "cozyfit", `rig-${character}.npz`);
|
|
328
|
+
}
|
|
329
|
+
|
|
330
|
+
function ensureTrackerRig(character, log) {
|
|
331
|
+
const path = trackerRigPath(character);
|
|
332
|
+
if (existsSync(path)) return path;
|
|
333
|
+
mkdirSync(dirname(path), { recursive: true });
|
|
334
|
+
const run = spawnSync(process.execPath, [join(ROOT, "tools/track/export-rig.mjs"), "--model", character, "--out", path], { cwd: ROOT, encoding: "utf8" });
|
|
335
|
+
if (run.status !== 0 || !existsSync(path)) throw new Error(`tracker rig export failed: ${run.stderr || run.stdout}`);
|
|
336
|
+
log(`T1: exported rig ${path}`);
|
|
337
|
+
return path;
|
|
338
|
+
}
|
|
339
|
+
|
|
340
|
+
export async function runT1({ item, inputs, itemDir, stepDir, options, endpoints, g5MotionPath, obsPath, log, trackerRunner = runTracker, masksRunner = ensureMasks } = {}) {
|
|
341
|
+
if (!g5MotionPath || !existsSync(g5MotionPath)) throw new Error("T1: G5 initializer is missing");
|
|
342
|
+
const rigPath = ensureTrackerRig(options.character, log);
|
|
343
|
+
const scenePath = inputs.scene ?? join(itemDir, "empty-scene.json");
|
|
344
|
+
if (!inputs.scene) writeFileSync(scenePath, "[]\n");
|
|
345
|
+
const endpointConfig = Object.fromEntries(endpoints.slice(0, 2).map((endpoint, index) => [index === 0 ? "a" : "b", {
|
|
346
|
+
rootPos: Array.from(endpoint.rootPos),
|
|
347
|
+
rotMats: Array.from(endpoint.rotMats),
|
|
348
|
+
}]));
|
|
349
|
+
const cameraPath = join(stepDir, "camera.json");
|
|
350
|
+
mkdirSync(stepDir, { recursive: true });
|
|
351
|
+
// Only this attempt's tracker diagnostics may describe the step.
|
|
352
|
+
rmSync(join(stepDir, "diagnostics.json"), { force: true });
|
|
353
|
+
writeFileSync(cameraPath, `${JSON.stringify({ ...readJson(inputs.camera), tracker: { ...(readJson(inputs.camera).tracker ?? {}), endpoints: endpointConfig } }, null, "\t")}\n`);
|
|
354
|
+
const attemptDir = join(stepDir, `tracker-${Date.now()}-${process.pid}`);
|
|
355
|
+
mkdirSync(attemptDir, { recursive: true });
|
|
356
|
+
try {
|
|
357
|
+
const sourceVideo = item.set === "fal" ? inputs.rawVideo : inputs.video;
|
|
358
|
+
const masks = await masksRunner({
|
|
359
|
+
video: sourceVideo,
|
|
360
|
+
obs: obsPath,
|
|
361
|
+
outDir: join(options.obsRoot ?? itemDir, item.set, item.name, "masks"),
|
|
362
|
+
host: options.host,
|
|
363
|
+
log: (line) => log(`T1 masks: ${line}`),
|
|
364
|
+
});
|
|
365
|
+
const deadline = Date.now() + options.gpuWaitMin * 60000;
|
|
366
|
+
let result;
|
|
367
|
+
for (;;) {
|
|
368
|
+
try {
|
|
369
|
+
result = await trackerRunner({
|
|
370
|
+
host: options.host,
|
|
371
|
+
video: sourceVideo,
|
|
372
|
+
obsPath,
|
|
373
|
+
masksPath: masks.path,
|
|
374
|
+
initMotionPath: g5MotionPath,
|
|
375
|
+
cameraPath,
|
|
376
|
+
scenePath,
|
|
377
|
+
rigPath: rigPath,
|
|
378
|
+
outDir: attemptDir,
|
|
379
|
+
ablate: options.trackAblate,
|
|
380
|
+
onLine: (line) => log(`T1: ${line}`),
|
|
381
|
+
});
|
|
382
|
+
break;
|
|
383
|
+
} catch (error) {
|
|
384
|
+
if (!/gpu-busy/.test(String(error.message)) || Date.now() > deadline) throw error;
|
|
385
|
+
log(`T1: GPU busy; waiting 30 s (${String(error.message).split("\\n").find((line) => /gpu-busy/.test(line))?.trim() ?? error.message})`);
|
|
386
|
+
await sleep(30000);
|
|
387
|
+
}
|
|
388
|
+
}
|
|
389
|
+
if (result.diagnostics?.failure) throw Object.assign(new Error(`tracker failure: ${result.diagnostics.failure}`), { diagnostics: result.diagnostics, diagnosticsPath: result.diagnosticsPath });
|
|
390
|
+
copyFileSync(result.motionPath, join(stepDir, "motion.npz"));
|
|
391
|
+
copyFileSync(result.diagnosticsPath, join(stepDir, "diagnostics.json"));
|
|
392
|
+
return { diagnostics: result.diagnostics, fallback: null, trackerError: null };
|
|
393
|
+
} catch (error) {
|
|
394
|
+
const step = fallbackStep(endpoints);
|
|
395
|
+
const fallbackDir = join(itemDir, step);
|
|
396
|
+
const fallbackMotion = join(fallbackDir, "motion.npz");
|
|
397
|
+
if (!existsSync(fallbackMotion)) throw new Error(`T1 tracker failed and ${step} motion is unavailable: ${error.message}`);
|
|
398
|
+
copyFileSync(fallbackMotion, join(stepDir, "motion.npz"));
|
|
399
|
+
const details = error instanceof AggregateError && error.errors?.length
|
|
400
|
+
? error.errors.map((cause) => cause?.message ?? String(cause)).join(" | ")
|
|
401
|
+
: error.message;
|
|
402
|
+
// A failed fit still reports why (e.g. keypoint-residual) beside the fallback.
|
|
403
|
+
const diagnostics = error.diagnostics && error.diagnosticsPath && existsSync(error.diagnosticsPath) ? error.diagnostics : null;
|
|
404
|
+
if (diagnostics) copyFileSync(error.diagnosticsPath, join(stepDir, "diagnostics.json"));
|
|
405
|
+
return { diagnostics, fallback: step, trackerError: { message: details, failure: diagnostics?.failure ?? null } };
|
|
406
|
+
} finally {
|
|
407
|
+
rmSync(attemptDir, { recursive: true, force: true });
|
|
408
|
+
}
|
|
409
|
+
}
|
|
410
|
+
|
|
411
|
+
async function runItem(item, options, commit) {
|
|
412
|
+
const itemDir = join(options.out, item.set, item.name);
|
|
413
|
+
mkdirSync(itemDir, { recursive: true });
|
|
414
|
+
const label = `${item.set}/${item.name}`;
|
|
415
|
+
const log = (line) => { const text = `[${new Date().toISOString()}] ${line}`; appendFileSync(join(itemDir, "bench.log"), `${text}\n`); console.log(`${label}: ${line}`); };
|
|
416
|
+
const inputs = itemInputs(item, itemDir);
|
|
417
|
+
const pending = options.steps.filter((step) => options.force || !readJson(join(itemDir, step, "result.json"))?.ok);
|
|
418
|
+
for (const step of options.steps.filter((s) => !pending.includes(s))) log(`${step}: skip (result.json ok; --force re-runs)`);
|
|
419
|
+
if (!pending.length) return { failed: 0 };
|
|
420
|
+
if (item.set === "fal") normalizeFalVideo(inputs, log);
|
|
421
|
+
const camera = readJson(inputs.camera);
|
|
422
|
+
const endpoints = readEndpoints(inputs.source);
|
|
423
|
+
const ankle = characterStandingAnkle(endpoints, options.character);
|
|
424
|
+
log(`character ankle plane ${ankle.height.toFixed(4)} m (${options.character}; A/B lower foot ${ankle.perPose.map((h) => h.toFixed(4)).join(", ")})`);
|
|
425
|
+
const ctx = { camera, ankleHeight: ankle.height, endpoints, boxes: sceneBoxes(inputs.scene).boxes };
|
|
426
|
+
const provenance = {};
|
|
427
|
+
const load = async (kind) => {
|
|
428
|
+
const obsRun = await ensureObs({ kind, inputs, itemDir, options, log });
|
|
429
|
+
const obs = readNpz(obsRun.path);
|
|
430
|
+
const rest = ensureRest(Array.from(obs.betas_used.data), options, log);
|
|
431
|
+
provenance[kind] = { obs: obsRun.path, obsSha256: fileSha(obsRun.path), manifest: obsRun.manifest, rest: rest.path, betasUsed: Array.from(obs.betas_used.data) };
|
|
432
|
+
return { obs, rest: rest.rest };
|
|
433
|
+
};
|
|
434
|
+
if (pending.some((s) => !MANNEQUIN_STEPS.has(s))) ctx.base = await load("base");
|
|
435
|
+
if (pending.some((s) => MANNEQUIN_STEPS.has(s))) ctx.mannequin = await load("mannequin");
|
|
436
|
+
const t1ObsPath = ctx.mannequin ? provenance.mannequin.obs : null;
|
|
437
|
+
let failed = 0;
|
|
438
|
+
for (const step of pending) {
|
|
439
|
+
const stepDir = join(itemDir, step), motionPath = join(stepDir, "motion.npz");
|
|
440
|
+
mkdirSync(stepDir, { recursive: true });
|
|
441
|
+
const started = performance.now();
|
|
442
|
+
const result = {
|
|
443
|
+
ok: false, tool: "tools/bench/obs-bench.mjs", commit, createdAt: new Date().toISOString(), item, step,
|
|
444
|
+
inputs: { camera: { path: inputs.camera, sha256: fileSha(inputs.camera) }, video: inputs.video, scene: inputs.scene, ankleHeightM: ctx.ankleHeight, ankle, endpoints: { path: inputs.source, frames: "first and last only" },
|
|
445
|
+
obs: MANNEQUIN_STEPS.has(step) ? provenance.mannequin : provenance.base },
|
|
446
|
+
};
|
|
447
|
+
try {
|
|
448
|
+
if (step === "T1") {
|
|
449
|
+
const g5Path = join(itemDir, "G5", "motion.npz");
|
|
450
|
+
const buildStep = ladderStep;
|
|
451
|
+
if (!existsSync(g5Path)) {
|
|
452
|
+
const g5 = buildStep("G5", ctx);
|
|
453
|
+
mkdirSync(dirname(g5Path), { recursive: true });
|
|
454
|
+
writeNpz(g5Path, motionArraysToNpzMembers(g5.motion));
|
|
455
|
+
}
|
|
456
|
+
const fallbackName = fallbackStep(endpoints), fallbackPath = join(itemDir, fallbackName, "motion.npz");
|
|
457
|
+
if (!existsSync(fallbackPath)) {
|
|
458
|
+
const fallbackBuilt = buildStep(fallbackName, ctx);
|
|
459
|
+
mkdirSync(dirname(fallbackPath), { recursive: true });
|
|
460
|
+
writeNpz(fallbackPath, motionArraysToNpzMembers(fallbackBuilt.motion));
|
|
461
|
+
}
|
|
462
|
+
const tracker = await runT1({ item, inputs, itemDir, stepDir, options, endpoints, g5MotionPath: g5Path, obsPath: t1ObsPath, log });
|
|
463
|
+
result.fallback = tracker.fallback;
|
|
464
|
+
result.trackAblate = options.trackAblate;
|
|
465
|
+
if (tracker.trackerError) result.trackerError = tracker.trackerError;
|
|
466
|
+
if (tracker.diagnostics) {
|
|
467
|
+
result.diagnostics = tracker.diagnostics;
|
|
468
|
+
result.runtimeS = tracker.diagnostics.runtime.trackerSeconds;
|
|
469
|
+
result.peakVramMB = tracker.diagnostics.runtime.peakReservedMiB;
|
|
470
|
+
result.lrSwitches = tracker.diagnostics.lrState.slice(1).filter((v, i) => v !== tracker.diagnostics.lrState[i]).length;
|
|
471
|
+
result.occludedFrames = tracker.diagnostics.occluded.filter((row) => row.some(Boolean)).length;
|
|
472
|
+
}
|
|
473
|
+
result.motion = { path: motionPath };
|
|
474
|
+
} else {
|
|
475
|
+
const built = ladderStep(step, ctx);
|
|
476
|
+
writeNpz(motionPath, motionArraysToNpzMembers(built.motion));
|
|
477
|
+
result.motion = { path: motionPath, frames: built.motion.frames, fps: built.motion.fps };
|
|
478
|
+
result.ladder = built.diagnostics;
|
|
479
|
+
}
|
|
480
|
+
log(`${step}: scoring`);
|
|
481
|
+
result.score = item.set === "fal" ? scoreFal({ item, inputs, motionPath, stepDir, options }) : scoreAgainstGt({ item, inputs, motionPath, stepDir, options });
|
|
482
|
+
result.ok = true;
|
|
483
|
+
log(`${step}: ok ${JSON.stringify(result.score, (k, v) => (typeof v === "number" ? Number(v.toFixed(4)) : v))}`);
|
|
484
|
+
} catch (error) {
|
|
485
|
+
failed += 1;
|
|
486
|
+
result.error = { message: error.message, stack: error.stack };
|
|
487
|
+
log(`${step}: FAILED ${error.message}`);
|
|
488
|
+
}
|
|
489
|
+
result.seconds = (performance.now() - started) / 1000;
|
|
490
|
+
putJson(join(stepDir, "result.json"), result);
|
|
491
|
+
}
|
|
492
|
+
return { failed };
|
|
493
|
+
}
|
|
494
|
+
|
|
495
|
+
export async function main(argv = process.argv.slice(2)) {
|
|
496
|
+
const options = parseArgs(argv);
|
|
497
|
+
if (options.help) { console.log(USAGE); return; }
|
|
498
|
+
const items = selectItems(readJson(options.approved), options.items);
|
|
499
|
+
const commit = commitId();
|
|
500
|
+
let failed = 0;
|
|
501
|
+
for (const item of items) {
|
|
502
|
+
try {
|
|
503
|
+
failed += (await runItem(item, options, commit)).failed;
|
|
504
|
+
} catch (error) {
|
|
505
|
+
failed += 1;
|
|
506
|
+
const itemDir = join(options.out, item.set, item.name);
|
|
507
|
+
mkdirSync(itemDir, { recursive: true });
|
|
508
|
+
appendFileSync(join(itemDir, "bench.log"), `[${new Date().toISOString()}] ITEM FAILED ${error.stack}\n`);
|
|
509
|
+
console.error(`${item.set}/${item.name}: ITEM FAILED ${error.message}`);
|
|
510
|
+
}
|
|
511
|
+
}
|
|
512
|
+
if (failed) {
|
|
513
|
+
console.error(`obs-bench: ${failed} failure(s)`);
|
|
514
|
+
process.exitCode = 1;
|
|
515
|
+
}
|
|
516
|
+
}
|
|
517
|
+
|
|
518
|
+
if (process.argv[1] && resolve(process.argv[1]) === fileURLToPath(import.meta.url)) {
|
|
519
|
+
main().catch((error) => { console.error(`obs-bench: ${error.stack ?? error}`); process.exitCode = 1; });
|
|
520
|
+
}
|