-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathrun_test_EGSCIT.sh
executable file
·95 lines (88 loc) · 2.34 KB
/
run_test_EGSCIT.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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
#!/bin/bash
source ~/.bashrc_liqianma
data_parent_dir='./data'
checkpoint_dir='./logs/'
gpu=0
segment_class=8
model=1
########################### Choose exp by uncomment ###########################
# data_name='mnist'
# dataset_dir='mnist_BW_test_28x28'
# bs_train=8
# bs_test=4
# num_style=1
# lr=1e-5
# n_1=1
# img_h=28
# img_w=28
# img_h_original=28
# img_w_original=28
# style_weight=1e3
# content_weight=1e1
# data_name='mnist_multi'
# dataset_dir='mnist_multi_jitterColor_BW_test_112x112'
# bs_train=8
# bs_test=4
# num_style=1
# lr=1e-5
# n_1=3
# img_h=112
# img_w=112
# img_h_original=112
# img_w_original=112
# style_weight=1e4
# content_weight=1e2
# data_name='gta25kcity'
# dataset_dir='gta25k_city_test_512x1024_8catId'
# bs_train=3
# bs_test=4
# num_style=1
# lr=1e-4
# n_1=3
# img_h=256
# img_w=512
# img_h_original=512
# img_w_original=1024
# style_weight=1e3
# content_weight=5e2
data_name='gta25kbdd'
dataset_dir='gta25k_bdd_test_512x1024_8catId'
bs_train=3
bs_test=4
num_style=1
lr=1e-4
n_1=3
img_h=256
img_w=512
img_h_original=512
img_w_original=1024
style_weight=1e4
content_weight=1e2
# data_name='celeba'
# dataset_dir='celebaMaleFemale_test_128x128'
# bs_train=8
# bs_test=1
# num_style=30
# lr=1e-4
# n_1=3
# img_h=128
# img_w=128
# img_h_original=128
# img_w_original=128
# style_weight=5e3
# content_weight=1e1
###############################################################
model_dir=${checkpoint_dir}'MODEL1_'${data_name}'_bs'${bs_train}'_lr'${lr}'_'${style_weight}'Style_'${content_weight}'Content'
pretrained_path=${checkpoint_dir}'MODEL0_'${data_name}'_bs'${bs_train}'_lr'${lr}'/UNIT.model-0'
test_model_path=${model_dir}'/UNIT.model-0'
python main.py --model_dir=${model_dir} --phase='test' \
--data_parent_dir=${data_parent_dir} --dataset_dir=${dataset_dir} \
--gpu=${gpu} --batch_size=${bs_test} --model=${model} \
--norm=None \
--epoch=1 --segment_class=${segment_class} \
--L1_weight=100 --L1_cycle_weight=100 --n_dis=4 --ngf=64 --ndf=64 \
--n_encoder=${n_1} --n_gen_decoder=${n_1} \
--img_h=${img_h} --img_w=${img_w} --img_h_original=${img_h_original} --img_w_original=${img_w_original} \
--test_model_path=${test_model_path} \
--pretrained_path=${pretrained_path} \
--num_style=${num_style} \