added fused classifier head

2026-03-02 06:44:22 +01:00
parent fd1452c187
commit 080b5999fa
11 changed files with 583 additions and 21 deletions
@@ -0,0 +1,45 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+# Bilateral tower runs (3 total) — UNet ROI crop:
+#   binary     × bilateral
+#   multiclass × bilateral
+#   multiclass × bilateral + balanced sampling
+
+ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/../../.." && pwd)"
+cd "$ROOT_DIR"
+
+CROP_ARGS=(
+  --img-crop-manifest manifest.csv
+  --img-crop-weights models/v2/refuge/segmentation/per_image_refuge_build/best.pt
+  --img-crop-normalize per_image
+)
+
+COMMON_ARGS=(
+  --epochs 40
+  --n-splits 5
+  --batch-size 8
+  --backbone refugelike
+  --tower-modes bilateral
+)
+
+# Runs 1-2: binary + multiclass bilateral (no balanced sampling)
+echo "[1/2] UNet ROI — binary + multiclass, bilateral..."
+python3 scripts/basic_analysis/compare_hypertower_modes.py \
+  "${COMMON_ARGS[@]}" \
+  "${CROP_ARGS[@]}" \
+  --eval-modes binary multiclass \
+  --run-name v2_modes_unet_40ep_5fold_bilateral_v2
+
+# Run 3: multiclass bilateral + balanced sampling
+# (reuse the crop cache built above)
+echo "[2/2] UNet ROI — multiclass, bilateral, balanced sampling..."
+python3 scripts/basic_analysis/compare_hypertower_modes.py \
+  "${COMMON_ARGS[@]}" \
+  "${CROP_ARGS[@]}" \
+  --eval-modes multiclass \
+  --balanced-sampling \
+  --persist-img-crop-cache \
+  --run-name v2_modes_unet_40ep_5fold_bilateral_balanced_v2
+
+echo "Bilateral UNet runs complete."