From 4d87a6d781c3f159d216f4cd9251e3d7bd63554f Mon Sep 17 00:00:00 2001 From: skal Date: Thu, 12 Feb 2026 12:08:22 +0100 Subject: CNN v2: storage buffer architecture foundation MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - Add binary weight format (header + layer info + packed f16) - New export_cnn_v2_weights.py for binary weight export - Single cnn_v2_compute.wgsl shader with storage buffer - Load weights in CNNv2Effect::load_weights() - Create layer compute pipeline with 5 bindings - Fast training config: 100 epochs, 3×3 kernels, 8→4→4 channels Next: Complete bind group creation and multi-layer compute execution --- scripts/train_cnn_v2_full.sh | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'scripts') diff --git a/scripts/train_cnn_v2_full.sh b/scripts/train_cnn_v2_full.sh index 4ddd9ac..383922d 100755 --- a/scripts/train_cnn_v2_full.sh +++ b/scripts/train_cnn_v2_full.sh @@ -12,8 +12,8 @@ INPUT_DIR="training/input" TARGET_DIR="training/target_2" CHECKPOINT_DIR="checkpoints" VALIDATION_DIR="validation_results" -EPOCHS=10000 -CHECKPOINT_EVERY=500 +EPOCHS=100 +CHECKPOINT_EVERY=5 BATCH_SIZE=16 # Patch-based training (default) @@ -25,8 +25,8 @@ DETECTOR="harris" # FULL_IMAGE="--full-image" # IMAGE_SIZE=256 -KERNEL_SIZES="1 3 5" -CHANNELS="16 8 4" +KERNEL_SIZES="3 3 3" +CHANNELS="8 4 4" echo "=== CNN v2 Complete Training Pipeline ===" echo "Input: $INPUT_DIR" @@ -98,7 +98,7 @@ mkdir -p "$VALIDATION_DIR" # Test first input image with checkpoints at intervals TEST_IMAGE="$INPUT_DIR/img_000.png" -CHECKPOINT_INTERVAL=1000 +CHECKPOINT_INTERVAL=5 echo " Processing checkpoints (every ${CHECKPOINT_INTERVAL} epochs)..." -- cgit v1.2.3