script editsHEADmaster

1 files changed, 35 insertions, 0 deletions

diff --git a/run_one.sh b/run_one.sh
new file mode 100755
index 0000000..e325367
--- /dev/null
+++ b/run_one.sh
@@ -0,0 +1,35 @@
+device='/gpu:0'
+
+# Parse arguments
+content_image="$1"
+content_dir=$(dirname "$content_image")
+content_filename=$(basename "$content_image")
+
+style_image="$2"
+style_dir=$(dirname "$style_image" )
+style_filename=$(basename "$style_image")
+
+echo "Rendering stylized image. This may take a while..."
+python neural_style.py \
+       --content_img "${content_filename}" \
+       --content_img_dir "${content_dir}" \
+       --style_imgs "${style_filename}" \
+       --style_imgs_dir "${style_dir}" \
+       --device "${device}" \
+       --verbose;
+
+#mkdir $3_color
+#cp image_output/result/* $3_color/
+#
+#python neural_style.py \
+#       --content_img "${content_filename}" \
+#       --content_img_dir "${content_dir}" \
+#       --style_imgs "${style_filename}" \
+#       --style_imgs_dir "${style_dir}" \
+#       --device "${device}" \
+#       --original_colors \
+#       --verbose;
+#
+mkdir $3
+cp image_output/result/* $3/
+cp image_output/result/result.png $3.png