-
Notifications
You must be signed in to change notification settings - Fork 1
/
multires_style2content_hist.sh
127 lines (102 loc) · 5.33 KB
/
multires_style2content_hist.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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
# Multiscale Generation With Histogram Matching
INTERPRETER=python3 # Replace with 'th' for neural_style.lua or 'python' for python 2
SCRIPT=neural_style.py # Replace with 'neural_style.lua' for the original neural-style
NEURAL_STYLE=$INTERPRETER
NEURAL_STYLE+=" "
NEURAL_STYLE+=$SCRIPT
# Uncomment if using pip package
#NEURAL_STYLE=neural-style
# Input Images
CONTENT_IMAGE=content_image.jpg
STYLE_IMAGE="style_image.png"
# Basic Parameters
CONTENT_WEIGHT=5
STYLE_WEIGHT=100
STYLE_SCALE=1
MODEL_FILE='models/vgg19-d01eb7cb.pth'
# Histogram Matching from style image to content image
python linear-color-transfer.py --target_image $CONTENT_IMAGE --source_image $STYLE_IMAGE --output_image content_colored_pca.png
$NEURAL_STYLE \
-content_image content_colored_pca.png \
-style_image $STYLE_IMAGE \
-init image \
-tv_weight 0 -seed 876 -save_iter 500 -print_iter 50 -backend cudnn -cudnn_autotune \
-content_weight $CONTENT_WEIGHT -style_weight $STYLE_WEIGHT -style_scale $STYLE_SCALE -model_file $MODEL_FILE \
-output_image out1.png \
-image_size 640 \
-num_iterations 1500
python linear-color-transfer.py --target_image out1.png --source_image $STYLE_IMAGE --output_image out1_hist_colored_pca.png
$NEURAL_STYLE \
-content_image content_colored_pca.png \
-style_image $STYLE_IMAGE \
-init image -init_image out1_hist_colored_pca.png \
-tv_weight 0 -seed 876 -save_iter 500 -print_iter 50 -backend cudnn -cudnn_autotune \
-content_weight $CONTENT_WEIGHT -style_weight $STYLE_WEIGHT -style_scale $STYLE_SCALE -model_file $MODEL_FILE \
-output_image out2.png \
-image_size 768 \
-num_iterations 1000
python linear-color-transfer.py --target_image out2.png --source_image $STYLE_IMAGE --output_image out2_hist_colored_pca.png
$NEURAL_STYLE \
-content_image content_colored_pca.png \
-style_image $STYLE_IMAGE \
-init image -init_image out2_hist_colored_pca.png \
-tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -backend cudnn -cudnn_autotune \
-content_weight $CONTENT_WEIGHT -style_weight $STYLE_WEIGHT -style_scale $STYLE_SCALE -model_file $MODEL_FILE \
-image_size 1024 \
-num_iterations 500 \
-output_image out3.png \
python linear-color-transfer.py --target_image out3.png --source_image $STYLE_IMAGE --output_image out3_hist_colored_pca.png
$NEURAL_STYLE \
-content_image content_colored_pca.png \
-style_image $STYLE_IMAGE \
-init image -init_image out3_hist_colored_pca.png \
-tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -backend cudnn -cudnn_autotune \
-content_weight $CONTENT_WEIGHT -style_weight $STYLE_WEIGHT -style_scale $STYLE_SCALE -model_file $MODEL_FILE \
-image_size 1152 \
-num_iterations 200 \
-output_image out4.png \
python linear-color-transfer.py --target_image out4.png --source_image $STYLE_IMAGE --output_image out4_hist_colored_pca.png
$NEURAL_STYLE \
-content_image content_colored_pca.png \
-style_image $STYLE_IMAGE \
-init image -init_image out4_hist_colored_pca.png \
-tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -backend cudnn -cudnn_autotune \
-content_weight $CONTENT_WEIGHT -style_weight $STYLE_WEIGHT -style_scale $STYLE_SCALE -model_file $MODEL_FILE \
-image_size 1536 \
-num_iterations 200 \
-output_image out5.png \
python linear-color-transfer.py --target_image out5.png --source_image $STYLE_IMAGE --output_image out5_hist_colored_pca.png
$NEURAL_STYLE \
-content_image content_colored_pca.png \
-style_image $STYLE_IMAGE \
-init image -init_image out5_hist_colored_pca.png \
-tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -backend cudnn -cudnn_autotune \
-content_weight $CONTENT_WEIGHT -style_weight $STYLE_WEIGHT -style_scale $STYLE_SCALE -model_file $MODEL_FILE \
-image_size 1664 \
-num_iterations 200 \
-output_image out6.png \
python linear-color-transfer.py --target_image out6.png --source_image $STYLE_IMAGE --output_image out6_hist_colored_pca.png
$NEURAL_STYLE \
-content_image content_colored_pca.png \
-style_image $STYLE_IMAGE \
-init image -init_image out6_hist_colored_pca.png \
-tv_weight 0 -seed 876 -save_iter 100 -print_iter 50 -backend cudnn -cudnn_autotune -optimizer adam \
-content_weight $CONTENT_WEIGHT -style_weight $STYLE_WEIGHT -style_scale $STYLE_SCALE -model_file $MODEL_FILE \
-image_size 1920 \
-num_iterations 200 \
-output_image out7.png
python linear-color-transfer.py --target_image out7.png --source_image $STYLE_IMAGE --output_image out7_hist_colored_pca.png
$NEURAL_STYLE \
-content_image content_colored_pca.png \
-style_image $STYLE_IMAGE \
-init image -init_image out7_hist_colored_pca.png \
-tv_weight 0 -seed 876 -save_iter 100 -print_iter 50 -backend cudnn -cudnn_autotune -optimizer adam \
-content_weight $CONTENT_WEIGHT -style_weight $STYLE_WEIGHT -style_scale $STYLE_SCALE -model_file $MODEL_FILE \
-image_size 2125 \
-num_iterations 200 \
-output_image out8.png
python linear-color-transfer.py --target_image out8.png --source_image $STYLE_IMAGE --output_image out8_hist_colored_pca.png
# Luminance Transfer from histogram matched content image, to the final output
python lum-transfer.py --output_lum2 out8.png --cp_mode lum2 --output_image out8_final.png --org_content content_colored_pca.png
# Luminance Transfer from unmodified content image, to the final output
python lum-transfer.py --output_lum2 out8.png --cp_mode lum2 --output_image out8_final_2.png --org_content $CONTENT_IMAGE