1 files changed, 16 insertions, 4 deletions
diff --git a/scripts/train_cnn_v2_full.sh b/scripts/train_cnn_v2_full.sh
index 119b788..4ddd9ac 100755
--- a/scripts/train_cnn_v2_full.sh
+++ b/scripts/train_cnn_v2_full.sh
@@ -14,8 +14,17 @@ CHECKPOINT_DIR="checkpoints"
 VALIDATION_DIR="validation_results"
 EPOCHS=10000
 CHECKPOINT_EVERY=500
-BATCH_SIZE=8
-IMAGE_SIZE=256
+BATCH_SIZE=16
+
+# Patch-based training (default)
+PATCH_SIZE=32
+PATCHES_PER_IMAGE=64
+DETECTOR="harris"
+
+# Full-image training (alternative - uncomment to use)
+# FULL_IMAGE="--full-image"
+# IMAGE_SIZE=256
+
 KERNEL_SIZES="1 3 5"
 CHANNELS="16 8 4"
 
@@ -31,13 +40,16 @@ echo "[1/4] Training CNN v2 model..."
 python3 training/train_cnn_v2.py \
   --input "$INPUT_DIR" \
   --target "$TARGET_DIR" \
-  --image-size $IMAGE_SIZE \
+  --patch-size $PATCH_SIZE \
+  --patches-per-image $PATCHES_PER_IMAGE \
+  --detector $DETECTOR \
   --kernel-sizes $KERNEL_SIZES \
   --channels $CHANNELS \
   --epochs $EPOCHS \
   --batch-size $BATCH_SIZE \
   --checkpoint-dir "$CHECKPOINT_DIR" \
-  --checkpoint-every $CHECKPOINT_EVERY
+  --checkpoint-every $CHECKPOINT_EVERY \
+  $FULL_IMAGE
 
 if [ $? -ne 0 ]; then
     echo "Error: Training failed"