summaryrefslogtreecommitdiff
path: root/go.sh
diff options
context:
space:
mode:
authorskal <pascal.massimino@gmail.com>2026-02-11 17:23:07 +0100
committerskal <pascal.massimino@gmail.com>2026-02-11 17:23:07 +0100
commit46e0935ba3b241dcd0e965e492ef8fa270b537ea (patch)
treea277ce8d7daa0ca6fa7cca7f757366a70767b7cc /go.sh
parent606a3e8027e901b5a3f9e68444d931982080bdd9 (diff)
update cnn code
Diffstat (limited to 'go.sh')
-rw-r--r--go.sh5
1 files changed, 5 insertions, 0 deletions
diff --git a/go.sh b/go.sh
index 1f2fc26..e6ad52b 100644
--- a/go.sh
+++ b/go.sh
@@ -3,3 +3,8 @@
# ./training/train_cnn.py --layers 3 --kernel_sizes 3,3,3 --epochs 10000 --batch_size 16 --input training/input/ --target training/target_2/ --checkpoint-every 1000
./training/train_cnn.py --export-only training/checkpoints/checkpoint_epoch_2000.pth
./training/train_cnn.py --export-only training/checkpoints/checkpoint_epoch_2000.pth --infer training/input/img_001.png --output test/toto.png
+./training/train_cnn.py --export-only training/checkpoints/checkpoint_epoch_2000.pth \
+ --infer training/input/img_001.png \
+ --output output/ref/toto0.png --save-intermediates output/ref/
+./build/cnn_test training/input/img_001.png output/toto.png --save-intermediates output/
+open output/*