1 files changed, 45 insertions, 0 deletions
diff --git a/training/debug/debug.sh b/training/debug/debug.sh
new file mode 100755
index 0000000..083082b
--- /dev/null
+++ b/training/debug/debug.sh
@@ -0,0 +1,45 @@
+#!/bin/sh
+
+pwd=`pwd`
+
+img=../input/img_003.png
+
+# img=/Users/skal/black_512x512_rgba.png
+#img=/Users/skal/rgba_0_0_0_0.png
+check_pt=../checkpoints/checkpoint_epoch_10000.pth
+#check_pt=../chk_5000_3x3x3.pt
+
+#../train_cnn.py --layers 3 --kernel_sizes 3,3,3 --epochs 10000 --batch_size 8 --input  ../input/ --target ../target_2/ --checkpoint-every 1000
+#../train_cnn.py --export-only ${check_pt}
+#../train_cnn.py --export-only ${check_pt} --infer ${img} --output test/toto.png
+
+#../train_cnn.py --layers 2 --kernel_sizes 1,1 --epochs 10 --batch_size 5 --input ../input/ --target ../target_2/ --checkpoint-every 10
+#../train_cnn.py --export-only ${check_pt}
+#../train_cnn.py --export-only ${check_pt} --infer ${img} --output test/toto.png
+
+## XXX uncomment!
+../train_cnn.py --export-only ${check_pt} \
+  --infer ${img} \
+  --output ref/toto.png --save-intermediates ref/ # --debug-hex
+
+echo "== GENERATE SHADERS =="
+echo
+cd ../../
+./training/train_cnn.py --export-only ${pwd}/${check_pt}
+
+echo "== COMPILE =="
+echo
+cmake --build build -j4 --target cnn_test
+cd ${pwd}
+
+echo "== RUN =="
+echo
+rm -f cur/toto.png
+../../build/cnn_test ${img} cur/toto.png --save-intermediates cur/ --layers 3 # --debug-hex
+
+open cur/*.png ref/*.png
+
+echo "open cur/*.png ref/*.png"
+
+#pngcrush -rem gAMA -rem sRGB cur/toto.png toto.png && mv toto.png cur/toto.png
+#pngcrush -rem gAMA -rem sRGB cur/layer_0.png toto.png && mv toto.png cur/layer_0.png