script editsHEADmaster

5 files changed, 93 insertions, 28 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..6bac78e
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,7 @@
+\#*
+*~
+*.wav
+*.png
+*.jpg
+*.mat
+
diff --git a/run_all.sh b/run_all.sh
new file mode 100755
index 0000000..c2478f5
--- /dev/null
+++ b/run_all.sh
@@ -0,0 +1,21 @@
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle1.tiff testf01
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle2.tiff testf02
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle3.tiff testf03
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle4.tiff testf04
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle5.tiff testf05
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle6.tiff testf06
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle7.tiff testf07
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle8.jpg testf08
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle9.jpg testf09
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/fakestyle/fakestyle10.jpg testf10
+
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style1.tiff test01
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style2.tiff test02
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style3.tiff test03
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style4.tiff test04
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style5.tiff test05
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style6.tiff test06
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style7.tiff test07
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style8.tiff test08
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style9.tiff test09
+./run_one.sh ../../data/tokyo_enhance/tokyo_short/thumb1000.jpg ../../Desktop/stylephotos/original/style10.tiff test10
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
diff --git a/stylize_image.sh b/stylize_image.sh
index 20e4ae9..fe9f707 100644..100755
--- a/stylize_image.sh
+++ b/stylize_image.sh
@@ -3,27 +3,27 @@ set -e
 cr=`echo $'\n.'`
 cr=${cr%.}
 
-if [ "$#" -le 1 ]; then
-   echo "Usage: bash stylize_image.sh <path_to_content_image> <path_to_style_image>"
-   exit 1
-fi
-
-echo ""
-read -p "Did you install the required dependencies? [y/n] $cr > " dependencies
-
-if [ "$dependencies" != "y" ]; then
-  echo "Error: Requires dependencies: tensorflow, opencv2 (python), scipy"
-  exit 1;
-fi
-
-echo ""
-read -p "Do you have a CUDA enabled GPU? [y/n] $cr > " cuda
-
-if [ "$cuda" != "y" ]; then
-  device='/cpu:0'
-else
+#if [ "$#" -le 1 ]; then
+#   echo "Usage: bash stylize_image.sh <path_to_content_image> <path_to_style_image>"
+#   exit 1
+#fi
+#
+#echo ""
+#read -p "Did you install the required dependencies? [y/n] $cr > " dependencies
+#
+#if [ "$dependencies" != "y" ]; then
+#  echo "Error: Requires dependencies: tensorflow, opencv2 (python), scipy"
+#  exit 1;
+#fi
+#
+#echo ""
+#read -p "Do you have a CUDA enabled GPU? [y/n] $cr > " cuda
+#
+#if [ "$cuda" != "y" ]; then
+#  device='/cpu:0'
+#else
   device='/gpu:0'
-fi
+#fi
 
 # Parse arguments
 content_image="$1"
@@ -41,4 +41,4 @@ python neural_style.py \
 --style_imgs "${style_filename}" \
 --style_imgs_dir "${style_dir}" \
 --device "${device}" \
---verbose;
\ No newline at end of file
+--verbose;
diff --git a/stylize_video.sh b/stylize_video.sh
index 5f39f89..bc2f8f7 100644..100755
--- a/stylize_video.sh
+++ b/stylize_video.sh
@@ -1,4 +1,4 @@
-set -e
+#set -e
 # Get a carriage return into `cr`
 cr=`echo $'\n.'`
 cr=${cr%.}
@@ -72,12 +72,14 @@ cd ..
 
 echo "Rendering stylized video frames [CPU & GPU]. This will take a while..."
 python neural_style.py --video \
---video_input_dir "${temp_dir}" \
---style_imgs_dir "${style_dir}" \
---style_imgs "${style_filename}" \
---end_frame "${num_frames}" \
---max_size "${max_size}" \
---verbose;
+       --video_input_dir "${temp_dir}" \
+       --style_imgs_dir "./" \
+       --style_imgs ../../Desktop/stylephotos/fakestyle/fakestyle6.tiff ../../Desktop/stylephotos/original/style5.tiff \
+       --style_imgs_weights 0.4 0.6 \
+       --end_frame "${num_frames}" \
+       --max_size "${max_size}" \
+       --verbose;
+
 
 # Create video from output images.
 echo "Converting image sequence to video.  This should be quick..."