#!/usr/bin/env bash
set -euo pipefail

# Binary runs v2.2 (4 total):
#   UNet crop:  single | fused head
#   GT crop:    single | fused head

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../../.." && pwd)"
cd "$ROOT_DIR"

MANIFEST="manifest.csv"
UNET_WEIGHTS="models/v2/refuge/segmentation/per_image/best.pt"

COMMON=(
  --epochs 40
  --n-splits 5
  --batch-size 8
  --backbone refugelike
  --eval-mode binary
  --single-warmup-tower-epochs 4
  --single-warmup-fused-epochs 4
  --img-crop-manifest "$MANIFEST"
)

UNET_CROP=(
  --img-crop-weights "$UNET_WEIGHTS"
)

GT_CROP=(
  --img-crop-gt
)

# ── UNet crop ────────────────────────────────────────────────────────────────

echo "[1/4] UNet crop — binary, single..."
python3 scripts/main/v2/run_multifold_v2.py \
  "${COMMON[@]}" "${UNET_CROP[@]}" \
  --tower-mode single \
  --run-name v2.2_single_binary_unet_40ep_5fold

echo "[2/4] UNet crop — binary, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
  "${COMMON[@]}" "${UNET_CROP[@]}" \
  --tower-mode ensemble \
  --fused-head --fusion-epochs 20 \
  --run-name v2.2_fused_binary_unet_40ep_5fold

# ── GT crop ──────────────────────────────────────────────────────────────────

echo "[3/4] GT crop — binary, single..."
python3 scripts/main/v2/run_multifold_v2.py \
  "${COMMON[@]}" "${GT_CROP[@]}" \
  --tower-mode single \
  --run-name v2.2_single_binary_gt_40ep_5fold

echo "[4/4] GT crop — binary, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
  "${COMMON[@]}" "${GT_CROP[@]}" \
  --tower-mode ensemble \
  --fused-head --fusion-epochs 20 \
  --run-name v2.2_fused_binary_gt_40ep_5fold

echo "Binary v2.2 runs complete."