forked from ZZUTK/SRNTT
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest.sh
27 lines (24 loc) · 786 Bytes
/
test.sh
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
#!/usr/bin/env bash
echo Demo of the testing process
# test on default SRNTT model
python main.py \
--is_train False \
--input_dir data/test/CUFED5/001_0.png \
--ref_dir data/test/CUFED5/001_2.png \
--use_init_model_only False \
--result_dir demo_testing_srntt
## test on default SRNTT-l2 model
#python main.py \
# --is_train False \
# --input_dir data/test/CUFED5/001_0.png \
# --ref_dir data/test/CUFED5/001_2.png \
# --use_init_model_only True \
# --result_dir demo_testing_srntt-l2
## test on your own model
#python main.py \
# --is_train False \
# --input_dir data/test/CUFED5/001_0.png \
# --ref_dir data/test/CUFED5/001_2.png \
# --use_init_model_only False \
# --result_dir demo_testing_yours \
# --save_dir your_model_dir