-
Notifications
You must be signed in to change notification settings - Fork 1
/
test_v4.sh
executable file
·112 lines (109 loc) · 3.56 KB
/
test_v4.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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#CUDA_VISIBLE_DEVICES=0 python test.py \
# --test_dataset "/media/glory/Transcend/Dataset/Scene/DIV2K_valid_HR" \
# --tgt_dataset "../art_dataset_v2" \
# --src_img_size 512 \
# --tgt_img_size 256 \
# --tgt_crop_size 256 \
# --workers 1 \
# --feature_dim 163840 \
# --gamma_dim 64 \
# --beta_dim 64 \
# --omega_dim 64 \
# --encoder_out_ch 64 \
# --db_number 8 \
# --ws 128 \
# --M 0 \
# --assigned_labels 6 \
# --sample_dir "train_log_v4/stylized_DIV2K_valid" \
# --save_extend ".png" \
# --pretrain_model "train_log_v4/checkpoint_DRB_GAN_edge_patch_style_data_v2/DRBGAN_it_499999.pt" \
# --mixture_list "LOUIS WAIN" "kaka" \
# --mixture_weights 0.5 0.5 \
# --sample_compared
#
#CUDA_VISIBLE_DEVICES=0 python test.py \
# --test_dataset "../../Animal_dataset/val/afhg_val/cat" \
# --tgt_dataset "../art_dataset_v2" \
# --tgt_img_size 256 \
# --tgt_crop_size 256 \
# --workers 1 \
# --feature_dim 163840 \
# --gamma_dim 64 \
# --beta_dim 64 \
# --omega_dim 64 \
# --encoder_out_ch 64 \
# --db_number 8 \
# --ws 128 \
# --M 0 \
# --assigned_labels 6 \
# --sample_dir "train_log_v4/stylized_afhg_val_cat" \
# --ref_style_dir "train_log_v4/style_ref_afhg_val_cat" \
# --stored_content_dir "train_log_v4/afhq_cat_val" \
# --save_extend ".png" \
# --pretrain_model "train_log_v4/checkpoint_DRB_GAN_edge_patch_style_data_v2/DRBGAN_it_499999.pt" \
# --mixture_list "LOUIS WAIN" "kaka" \
# --mixture_weights 0.5 0.5
#CUDA_VISIBLE_DEVICES=0 python test.py \
# --test_dataset "../Stable_Diffusion_Demo" \
# --tgt_dataset "../art_dataset_v2" \
# --tgt_img_size 256 \
# --tgt_crop_size 256 \
# --workers 1 \
# --feature_dim 163840 \
# --gamma_dim 64 \
# --beta_dim 64 \
# --omega_dim 64 \
# --encoder_out_ch 64 \
# --db_number 8 \
# --ws 128 \
# --M 0 \
# --assigned_labels 6 \
# --sample_dir "train_log_v4/stylized_Stable_Diffusion_Demo" \
# --save_extend ".png" \
# --pretrain_model "train_log_v4/checkpoint_DRB_GAN_edge_patch_style_data_v2/DRBGAN_it_499999.pt" \
# --mixture_list "LOUIS WAIN" "kaka" \
# --mixture_weights 0.5 0.5 \
# --sample_compared
#CUDA_VISIBLE_DEVICES=0 python test.py \
# --test_dataset "../Stable_Diffusion_Demo" \
# --tgt_dataset "../art_dataset_v2" \
# --tgt_img_size 256 \
# --tgt_crop_size 256 \
# --workers 1 \
# --feature_dim 163840 \
# --gamma_dim 64 \
# --beta_dim 64 \
# --omega_dim 64 \
# --encoder_out_ch 64 \
# --db_number 8 \
# --ws 128 \
# --M 0 \
# --assigned_labels 6 \
# --sample_dir "train_log_v4/stylized_Stable_Diffusion_Demo_add_random_noise" \
# --save_extend ".png" \
# --pretrain_model "train_log_v4/checkpoint_DRB_GAN_edge_patch_style_data_v2/DRBGAN_it_499999.pt" \
# --mixture_list "LOUIS WAIN" "kaka" \
# --mixture_weights 0.5 0.5 \
# --add_random_noise
CUDA_VISIBLE_DEVICES=0 python test.py \
--test_dataset "../Stable_Diffusion_Demo" \
--tgt_dataset "../art_dataset_v2" \
--tgt_img_size 256 \
--tgt_crop_size 256 \
--workers 1 \
--feature_dim 163840 \
--gamma_dim 64 \
--beta_dim 64 \
--omega_dim 64 \
--encoder_out_ch 64 \
--db_number 8 \
--ws 128 \
--M 0 \
--assigned_labels 6 \
--sample_dir "train_log_v4/stylized_Stable_Diffusion_Demo_use_sharpen" \
--save_extend ".png" \
--pretrain_model "train_log_v4/checkpoint_DRB_GAN_edge_patch_style_data_v2/DRBGAN_it_499999.pt" \
--mixture_list "LOUIS WAIN" "kaka" \
--mixture_weights 0.5 0.5 \
--sample_compared \
--use_sharpen