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

# Pick one downloaded sheet/candidate file first.
SHEET_ID="${1:-K-34-009-2}"
TIF="data/raw/${SHEET_ID}/${SHEET_ID}.tif"
CAND="data/interim/candidates/${SHEET_ID}_candidates.csv"
OUT="data/yolo/${SHEET_ID}"

python -m bgtopo_poc.cli export-yolo \
  --config configs/blue_detector.yaml \
  --sheet-id "$SHEET_ID" \
  --tif "$TIF" \
  --candidates "$CAND" \
  --out-dir "$OUT" \
  --tile-size 1024 \
  --overlap 128

# RTX 3080 FE 10 GB: start batch 2-4 at imgsz=1024. Raise only if VRAM allows it.
python -m bgtopo_poc.cli train-yolo \
  --data-yaml "$OUT/data.yaml" \
  --model yolov8s.pt \
  --imgsz 1024 \
  --epochs 80 \
  --batch 4 \
  --device 0