predictions class added

This commit is contained in:
rpotter6298
2026-03-04 11:17:10 +01:00
parent d36b9508ff
commit 13ad32683f
4 changed files with 338 additions and 49 deletions
+11 -22
View File
@@ -1,9 +1,9 @@
#!/usr/bin/env bash
set -euo pipefail
# Binary runs v2.2 (6 total):
# UNet crop: single | ensemble | fused head
# GT crop: single | ensemble | fused head
# 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"
@@ -17,6 +17,8 @@ COMMON=(
--batch-size 8
--backbone refugelike
--eval-mode binary
--single-warmup-tower-epochs 4
--single-warmup-fused-epochs 4
--img-crop-manifest "$MANIFEST"
)
@@ -30,45 +32,32 @@ GT_CROP=(
# ── UNet crop ────────────────────────────────────────────────────────────────
echo "[1/6] UNet crop — binary, single..."
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/6] UNet crop — binary, ensemble..."
echo "[2/4] UNet crop — binary, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${UNET_CROP[@]}" \
--tower-mode ensemble \
--run-name v2.2_ensemble_binary_unet_40ep_5fold
echo "[3/6] UNet crop — binary, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${UNET_CROP[@]}" \
--tower-mode ensemble \
--fused-head --fusion-epochs 10 \
--fused-head --fusion-epochs 20 \
--run-name v2.2_fused_binary_unet_40ep_5fold
# ── GT crop ──────────────────────────────────────────────────────────────────
echo "[4/6] GT crop — binary, single..."
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 "[5/6] GT crop — binary, ensemble..."
echo "[4/4] GT crop — binary, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${GT_CROP[@]}" \
--tower-mode ensemble \
--run-name v2.2_ensemble_binary_gt_40ep_5fold
echo "[6/6] GT crop — binary, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${GT_CROP[@]}" \
--tower-mode ensemble \
--img-crop-gt \
--fused-head --fusion-epochs 10 \
--fused-head --fusion-epochs 20 \
--run-name v2.2_fused_binary_gt_40ep_5fold
echo "Binary v2.2 runs complete."
+11 -21
View File
@@ -1,9 +1,9 @@
#!/usr/bin/env bash
set -euo pipefail
# Multiclass runs v2.2 (6 total):
# UNet crop: single | ensemble | fused head
# GT crop: single | ensemble | fused head
# Multiclass 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"
@@ -17,6 +17,8 @@ COMMON=(
--batch-size 8
--backbone refugelike
--eval-mode multiclass
--single-warmup-tower-epochs 4
--single-warmup-fused-epochs 4
--img-crop-manifest "$MANIFEST"
)
@@ -30,44 +32,32 @@ GT_CROP=(
# ── UNet crop ────────────────────────────────────────────────────────────────
echo "[1/6] UNet crop — multiclass, single..."
echo "[1/4] UNet crop — multiclass, single..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${UNET_CROP[@]}" \
--tower-mode single \
--run-name v2.2_single_multiclass_unet_40ep_5fold
echo "[2/6] UNet crop — multiclass, ensemble..."
echo "[2/4] UNet crop — multiclass, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${UNET_CROP[@]}" \
--tower-mode ensemble \
--run-name v2.2_ensemble_multiclass_unet_40ep_5fold
echo "[3/6] UNet crop — multiclass, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${UNET_CROP[@]}" \
--tower-mode ensemble \
--fused-head --fusion-epochs 10 \
--fused-head --fusion-epochs 20 \
--run-name v2.2_fused_multiclass_unet_40ep_5fold
# ── GT crop ──────────────────────────────────────────────────────────────────
echo "[4/6] GT crop — multiclass, single..."
echo "[3/4] GT crop — multiclass, single..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${GT_CROP[@]}" \
--tower-mode single \
--run-name v2.2_single_multiclass_gt_40ep_5fold
echo "[5/6] GT crop — multiclass, ensemble..."
echo "[4/4] GT crop — multiclass, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${GT_CROP[@]}" \
--tower-mode ensemble \
--run-name v2.2_ensemble_multiclass_gt_40ep_5fold
echo "[6/6] GT crop — multiclass, ensemble + fused head..."
python3 scripts/main/v2/run_multifold_v2.py \
"${COMMON[@]}" "${GT_CROP[@]}" \
--tower-mode ensemble \
--fused-head --fusion-epochs 10 \
--fused-head --fusion-epochs 20 \
--run-name v2.2_fused_multiclass_gt_40ep_5fold
echo "Multiclass v2.2 runs complete."