blob: 083082bd60d7625746cf76a3318bde5467edd5be (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
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
|
