Add files via upload

ProGamerGov · Oct 12, 2019 · 7fd8127 · 7fd8127
1 parent 90512bd
commit 7fd8127
Showing 1 changed file with 32 additions and 22 deletions.
diff --git a/multires_style2content_large.sh b/multires_style2content_large.sh
@@ -3,13 +3,23 @@
 CONTENT_IMAGE=content_image.jpg
 STYLE_IMAGE=style_image.png
 
-INTERPRETER=th
-NEURAL_STYLE=neural_style.lua
+INTERPRETER=th # Replace with python/python3 for neural-style-pt
+SCRIPT=neural_style.lua # Replace with neural_style.py for neural-style-pt
+
+NEURAL_STYLE=$INTERPRETER
+NEURAL_STYLE+=" "
+NEURAL_STYLE+=$SCRIPT
+
+# Uncomment if using pip package
+#NEURAL_STYLE=neural-style
+
+
+MULTIDEVICE_PARAMETER="-multigpu_strategy" # Replace -multigpu_strategy with -multidevice_strategy for neural-style-pt
 
 #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
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image \
@@ -20,7 +30,7 @@ $INTERPRETER $NEURAL_STYLE \
 
 python linear-color-transfer.py --target_image out1.png --source_image $STYLE_IMAGE --output_image out1_hist_colored_pca.png
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out1_hist_colored_pca.png \
@@ -31,7 +41,7 @@ $INTERPRETER $NEURAL_STYLE \
 
 python linear-color-transfer.py --target_image out2.png --source_image $STYLE_IMAGE --output_image out2_hist_colored_pca.png
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out2_hist_colored_pca.png \
@@ -42,7 +52,7 @@ $INTERPRETER $NEURAL_STYLE \
 
 python linear-color-transfer.py --target_image out3.png --source_image $STYLE_IMAGE --output_image out3_hist_colored_pca.png
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out3_hist_colored_pca.png \
@@ -53,88 +63,88 @@ $INTERPRETER $NEURAL_STYLE \
 
 python linear-color-transfer.py --target_image out4.png --source_image $STYLE_IMAGE --output_image out4_hist_colored_pca.png
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out4_hist_colored_pca.png \
-  -gpu 0,1,2,3,4,5,6,7 -multigpu_strategy 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune \
+  -gpu 0,1,2,3,4,5,6,7 $MULTIDEVICE_PARAMETER 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune \
   -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
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out5_hist_colored_pca.png \
-  -gpu 0,1,2,3,4,5,6,7 -multigpu_strategy 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune \
+  -gpu 0,1,2,3,4,5,6,7 $MULTIDEVICE_PARAMETER 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune \
   -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
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out6_hist_colored_pca.png \
-  -gpu 0,1,2,3,4,5,6,7 -multigpu_strategy 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune \
+  -gpu 0,1,2,3,4,5,6,7 $MULTIDEVICE_PARAMETER 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune \
   -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
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out7_hist_colored_pca.png \
-  -gpu 0,1,2,3,4,5,6,7 -multigpu_strategy 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
+  -gpu 0,1,2,3,4,5,6,7 $MULTIDEVICE_PARAMETER 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
   -image_size 2048 \
   -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
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out8_hist_colored_pca.png \
-  -gpu 0,1,2,3,4,5,6,7 -multigpu_strategy 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
+  -gpu 0,1,2,3,4,5,6,7 $MULTIDEVICE_PARAMETER 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
   -image_size 2432 \
   -num_iterations 200 \
   -output_image out9.png
 
 python linear-color-transfer.py --target_image out9.png --source_image $STYLE_IMAGE --output_image out9_hist_colored_pca.png
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out9_hist_colored_pca.png \
-  -gpu 0,1,2,3,4,5,6,7 -multigpu_strategy 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
+  -gpu 0,1,2,3,4,5,6,7 $MULTIDEVICE_PARAMETER 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
   -image_size 2560 \
   -num_iterations 200 \
   -output_image out10.png
 
 python linear-color-transfer.py --target_image out10.png --source_image $STYLE_IMAGE --output_image out10_hist_colored_pca.png
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out10_hist_colored_pca.png \
-  -gpu 0,1,2,3,4,5,6,7 -multigpu_strategy 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
+  -gpu 0,1,2,3,4,5,6,7 $MULTIDEVICE_PARAMETER 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
   -image_size 2816 \
   -num_iterations 200 \
   -output_image out11.png
 
 python linear-color-transfer.py --target_image out11.png --source_image $STYLE_IMAGE --output_image out11_hist_colored_pca.png
 
-$INTERPRETER $NEURAL_STYLE \
+$NEURAL_STYLE \
   -content_image content_colored_pca.png \
   -style_image $STYLE_IMAGE \
   -init image -init_image out11_hist_colored_pca.png \
-  -gpu 0,1,2,3,4,5,6,7 -multigpu_strategy 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
+  -gpu 0,1,2,3,4,5,6,7 $MULTIDEVICE_PARAMETER 3,6,12,15,20,26,31 -tv_weight 0 -seed 876 -save_iter 0 -print_iter 50 -init image -backend cudnn -cudnn_autotune -optimizer adam \
   -image_size 2944 \
   -num_iterations 200 \
   -output_image out12.png