summaryrefslogtreecommitdiff
path: root/go.sh
diff options
context:
space:
mode:
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/*