run.sh

#!/usr/bin/env bash
# Run inference: ./run.sh [task] [image(s)]
#   task: face_landmarker (default) | face_detector | face_decode | compare | face_decode_embed | compare_embed
#   image: path(s); for compare use two paths: ./run.sh compare img1.jpg img2.jpg
# Ensure you have run: ./install.sh
set -e

# --- Paths ---
ROOT="$(cd "$(dirname "$0")" && pwd)"
cd "$ROOT"
PYTHON="${ROOT}/.venv/bin/python"

# --- Check venv ---
if [ ! -x "$PYTHON" ]; then
  echo "Run ./install.sh first." >&2
  exit 1
fi

# --- Parse task and image ---
TASK="face_landmarker"
IMAGE=""
case "${1:-}" in
  face_landmarker|face_detector|face_decode|compare|face_decode_embed|compare_embed)
    TASK="$1"
    shift
    IMAGE="$*"
    ;;
  *)
    IMAGE="${1:-}"
    ;;
esac

# --- Resolve image: use first in images/ if none given (not for compare) ---
if [ -z "$IMAGE" ] && [ "$TASK" != "compare" ] && [ "$TASK" != "compare_embed" ]; then
  FIRST="$(find images -maxdepth 1 -type f \( -name '*.jpg' -o -name '*.jpeg' -o -name '*.png' \) 2>/dev/null | head -1)"
  if [ -z "$FIRST" ]; then
    echo "Usage: ./run.sh [task] [path/to/image.jpg ...]" >&2
    echo "  compare/compare_embed need two images: ./run.sh compare img1.jpg img2.jpg" >&2
    exit 1
  fi
  IMAGE="$FIRST"
fi

if [ "$TASK" = "compare" ] || [ "$TASK" = "compare_embed" ]; then
  if [ -z "$IMAGE" ]; then
    echo "Usage: ./run.sh $TASK path/to/img1.jpg path/to/img2.jpg" >&2
    exit 1
  fi
fi

# --- Run ---
if [ "$TASK" = "face_landmarker" ]; then
  "$PYTHON" run.py --task face_landmarker --no-blendshapes --image $IMAGE
elif [ "$TASK" = "face_detector" ]; then
  "$PYTHON" run.py --task face_detector --image $IMAGE
elif [ "$TASK" = "face_decode" ]; then
  "$PYTHON" run.py --task face_decode --image $IMAGE
elif [ "$TASK" = "compare" ]; then
  "$PYTHON" run.py --task compare --image $IMAGE
elif [ "$TASK" = "face_decode_embed" ]; then
  "$PYTHON" run.py --task face_decode_embed --image $IMAGE
elif [ "$TASK" = "compare_embed" ]; then
  "$PYTHON" run.py --task compare_embed --image $IMAGE
fi