#!/usr/bin/env bash
set -euo pipefail
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/../../.." && pwd)"
OUT="$ROOT/tools/keypose_image_sampler/out/v78_dynamic_warmup_all_mediapipe_final"
IMG="tools/keypose_image_sampler/external_pose_images_dynamic_warmup_v78_all_mediapipe_final"
MP_OUT="$OUT/mediapipe_candidates"
mkdir -p "$MP_OUT"
export PYTHONPATH="${NEXT11_MEDIAPIPE_PYDEPS:-/mnt/data/pydeps_mediapipe}:${PYTHONPATH:-}"
export LD_LIBRARY_PATH="${NEXT11_MEDIAPIPE_NATIVE_LIBS:-/mnt/data/native_libs}:/usr/lib/chromium:${LD_LIBRARY_PATH:-}"
cd "$ROOT"
python3 tools/keypose_image_sampler/mediapipe_keypose_images_to_ik_candidates.py \
  --image-root "$IMG" \
  --pose-model tools/keypose_image_sampler/models/pose_landmarker_full.task \
  --source-plan "$OUT/V78_ALL_MEDIAPIPE_SOURCE_PLAN_ADAPTER.json" \
  --out "$MP_OUT" \
  > "$OUT/V78_MEDIAPIPE_STDOUT.txt" \
  2> "$OUT/V78_MEDIAPIPE_STDERR.txt"
python3 tools/keypose_image_sampler/v78_scripts/build_v78_dynamic_warmup_all_mediapipe_final.py