rnd-v1
This commit is contained in:
26
scripts/run_training_example.sh
Normal file
26
scripts/run_training_example.sh
Normal file
@@ -0,0 +1,26 @@
|
||||
#!/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
|
||||
Reference in New Issue
Block a user