massive changes in the pipeline and ease of use

1 files changed, 13 insertions, 6 deletions

diff --git a/stylize.sh b/stylize.sh
index 0095629..434c079 100755
--- a/stylize.sh
+++ b/stylize.sh
@@ -1,18 +1,25 @@
-#!/bin/bash
+#!/bin/zsh
 
 ### You must be in the 'neural-art' directory when you run this
 
-if [ $# -eq 2 ]
+if [ $# -eq 0 ]
   then
-    echo "Video file needs to be specified"
+    echo "Style Image and Content Image need to be specified as arguments"
+    echo "Example: ./stylize.sh Images/Jitter_Doll.jpg Images/cute.jpg"
+    exit 1
+fi
+
+if [ $# -eq 1 ]
+  then
+    echo "Content image needs to be specified"
     exit 1
 fi
 
 # stylize data [pair (style, content)]
 python neuralart.py $1 $2
 
-# render images (actual frames) from images.npy
-python renderer.py
+# render images (actual frames) from (an) images.npy
+python renderer.py $1 $2
 
 # turn everything into a video
-ffmpeg -framerate 60 -pattern_type glob -i 'Output/neural_art_*.png' -vf "pad=ceil(iw/2)*2:ceil(ih/2)*2" $3
+ffmpeg -framerate 60 -pattern_type glob -i 'Output/neural_art_*.png' -vf "pad=ceil(iw/2)*2:ceil(ih/2)*2" ${${2/*\/}/.*}_in_${${1/*\/}/.*}.mp4