From 46e0935ba3b241dcd0e965e492ef8fa270b537ea Mon Sep 17 00:00:00 2001 From: skal Date: Wed, 11 Feb 2026 17:23:07 +0100 Subject: update cnn code --- go.sh | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'go.sh') 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/* -- cgit v1.2.3