nbrightandClaude Opus 5
609d9a6972
Reproduces the SUM Parts (CVPR 2025) face-labeling benchmark on a single consumer GPU, then applies it to drone-photogrammetry road survey meshes. Verified on RTX 3060 12GB / WSL2 Ubuntu 22.04 / CUDA 11.8 / torch 2.0.1: - CUDA extensions build (pointnet2_batch, pointops, chamfer_dist, emd, subsampling) - PointNet 100 epochs reaches mIoU 17.19, matching the paper's reported 15.1 - OBJ -> PLY conversion round-trips through the model and yields per-point predictions Four upstream source patches, all idempotent, originals preserved: - numpy aliases removed in 1.24 (np.long etc.) and collections ABCs moved in python 3.10 - the blind test split ships label = -1, which crashed ConfusionMatrix - mode=val referenced `epoch` before assignment Documents the traps that cost the most time, including VRAM overflow silently falling back to host RAM on WSL2 (25-100x slowdown, no OOM) and the colour scale mismatch between r/g/b float32 and red/green/blue uint8. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
35 lines
1.0 KiB
Bash
35 lines
1.0 KiB
Bash
#!/usr/bin/env bash
|
|
# SUM Parts - look at the actual epoch lines, to check for interleaving or
|
|
# a non-monotonic "best val miou"
|
|
set -uo pipefail
|
|
|
|
RUNROOT="$HOME/sum-parts/runs"
|
|
D="${1:-$(cat "$RUNROOT/.latest" 2>/dev/null)}"
|
|
LOG="$D/train.log"
|
|
|
|
echo "run: $D"
|
|
[ -f "$LOG" ] || { echo " no train.log"; exit 1; }
|
|
|
|
echo
|
|
echo "=== last 14 epoch lines ==="
|
|
grep -aE 'Epoch [0-9]+ LR' "$LOG" | tail -14
|
|
|
|
echo
|
|
echo "=== 'Find a better ckpt' events (last 8) ==="
|
|
grep -aE 'Find a better ckpt' "$LOG" | tail -8
|
|
|
|
echo
|
|
echo "=== how many distinct run_names wrote to this log ==="
|
|
grep -aoE 'sumv2_triangle-train-pointnet-ngpus1-[0-9]{8}-[0-9]{6}-[A-Za-z0-9]+' "$LOG" \
|
|
| sort -u | sed 's/^/ /'
|
|
|
|
echo
|
|
echo "=== effective val_freq (Val: bars per epoch) ==="
|
|
echo " epoch lines : $(grep -acE 'Epoch [0-9]+ LR' "$LOG")"
|
|
echo " val passes : $(grep -aoc 'Val: 100%' "$LOG" 2>/dev/null || echo '?')"
|
|
|
|
echo
|
|
echo "=== processes ==="
|
|
pgrep -af 'main.py --cfg' | awk '{print " pid " $1}' | head
|
|
echo " count: $(pgrep -cf 'main.py --cfg')"
|