27 lines
690 B
Bash
27 lines
690 B
Bash
#!/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
|