jcjohnson · jonmcoe · Apr 5, 2017 · Apr 5, 2017 · Apr 5, 2017 · Apr 5, 2017
diff --git a/.gitignore b/.gitignore
@@ -1,8 +1,19 @@
 *.swp
 out*.png
-*.png
-*.jpg
 *.prototxt*
 *.caffemodel
 models/
 !models/download_models.sh
+
+
+output/*.jpg
+output/final/*.jpg
+output/gif/*.gif
+content_incoming
+!content_incoming/.NOTE.md
+content_complete
+!content_complete/.NOTE.md
+all_styles
+!all_styles/.NOTE.md
+active_styles
+!active_styles/.NOTE.md
diff --git a/active_styles/.NOTE.md b/active_styles/.NOTE.md
@@ -0,0 +1 @@
+Style images in this directory (or symlinked here) will be applied
diff --git a/all_content_incoming.sh b/all_content_incoming.sh
@@ -0,0 +1 @@
+for file in content_incoming/*; do sh one_content_multi_styles.sh "$(basename $file)"; done
diff --git a/all_styles/.NOTE.md b/all_styles/.NOTE.md
@@ -0,0 +1 @@
+Keep all styles here
diff --git a/all_styles/alex_gray_diamond_being.jpg b/all_styles/alex_gray_diamond_being.jpg
diff --git a/all_styles/autumn_leaves.jpg b/all_styles/autumn_leaves.jpg
diff --git a/all_styles/batik.jpg b/all_styles/batik.jpg
diff --git a/all_styles/bryce_hudson.jpg b/all_styles/bryce_hudson.jpg
diff --git a/all_styles/drseusssketch.jpg b/all_styles/drseusssketch.jpg
diff --git a/all_styles/escher_sphere.jpg b/all_styles/escher_sphere.jpg
diff --git a/all_styles/frida_kahlo.jpg b/all_styles/frida_kahlo.jpg
diff --git a/all_styles/guernica.jpg b/all_styles/guernica.jpg
diff --git a/all_styles/kiddyppumpkins.jpg b/all_styles/kiddyppumpkins.jpg
diff --git a/all_styles/manet_flowers.jpg b/all_styles/manet_flowers.jpg
diff --git a/all_styles/marijuana_collage.jpg b/all_styles/marijuana_collage.jpg
diff --git a/all_styles/picasso_selfport1907.jpg b/all_styles/picasso_selfport1907.jpg
diff --git a/all_styles/pollock.jpg b/all_styles/pollock.jpg
diff --git a/all_styles/renandstimpy.jpg b/all_styles/renandstimpy.jpg
diff --git a/all_styles/seated_nude.jpg b/all_styles/seated_nude.jpg
diff --git a/all_styles/seurat_grand_jatte.jpg b/all_styles/seurat_grand_jatte.jpg
diff --git a/all_styles/shipwreck.jpg b/all_styles/shipwreck.jpg
diff --git a/all_styles/snow_scene.jpg b/all_styles/snow_scene.jpg
diff --git a/all_styles/spring_flowers.jpg b/all_styles/spring_flowers.jpg
diff --git a/all_styles/starry_night.jpg b/all_styles/starry_night.jpg
diff --git a/all_styles/summer_beach.jpg b/all_styles/summer_beach.jpg
diff --git a/all_styles/the_scream.jpg b/all_styles/the_scream.jpg
diff --git a/all_styles/warhol_monroe.jpg b/all_styles/warhol_monroe.jpg
diff --git a/all_styles/woman_with_hat_matisse.jpg b/all_styles/woman_with_hat_matisse.jpg
diff --git a/content_complete/.NOTE.md b/content_complete/.NOTE.md
@@ -0,0 +1 @@
+Content images are moved here after being styled
diff --git a/content_incoming/.NOTE.md b/content_incoming/.NOTE.md
@@ -0,0 +1 @@
+Files here will be pulled by shell script
diff --git a/make_gif.py b/make_gif.py
@@ -0,0 +1,34 @@
+import argparse
+import glob
+
+import images2gif
+from PIL import Image
+
+
+def main():
+    parser = argparse.ArgumentParser(description='Start and end for primary key')
+    parser.add_argument('--direction', '-d', type=str, nargs='?', default='backwards')
+    parser.add_argument('--prefix', '-p', type=str)
+    parser.add_argument('--output', '-o', type=str, nargs='?', default=None)
+    parser.add_argument('--original', '-r', type=str, nargs='?', default=None)
+
+    namespace = parser.parse_args()
+    prefix = namespace.prefix
+    original = namespace.original
+    direction = namespace.direction
+    output = namespace.output or 'output/gifs/{0}.gif'.format(prefix.split('/')[-1]+'_'+direction)
+
+    images = [Image.open(i) for i in glob.glob(prefix+'*')]
+    images = images[1:] + [images[0]]  # hack bc final iteration has no number
+    if original:
+        original_raw = Image.open(original)
+        original_resized = original_raw.resize(images[0].size)
+        images.append(original_resized)
+    # name reflects iteration direction. backwards is normal to weird. forwards is weird to normal
+    if direction == 'backwards':
+        images = images[::-1]
+
+    images2gif.writeGif(output, images, duration=.2)  # TODO: maybe pause at the end? use duration=[list] to change
+
+if __name__ == '__main__':
+    main()
diff --git a/one_content_multi_styles.sh b/one_content_multi_styles.sh
@@ -0,0 +1,19 @@
+#!/bin/bash
+SCRIPT=$0
+CONTENT_IMAGE=$1
+STYLES_DIR=active_styles
+CONTENT_INCOMING_DIR=content_incoming
+CONTENT_COMPLETE_DIR=content_complete
+
+for i in $( ls $STYLES_DIR); do
+    DEST_IMAGE=output/${CONTENT_IMAGE%????}__$i
+    if [ ! -f "$DEST_IMAGE" ]; then
+      echo attempting to apply $STYLES_DIR/$i to $CONTENT_IMAGE and save into $DEST_IMAGE
+      th neural_style.lua -gpu 0 -backend cudnn -style_image $STYLES_DIR/$i -content_image $CONTENT_INCOMING_DIR/$CONTENT_IMAGE -output_image $DEST_IMAGE
+    else
+      echo "skipping bc $DEST_IMAGE already exists"
+    fi
+    # python make_gif.py -p ${DEST_IMAGE%????} -r $CONTENT_INCOMING_DIR/$CONTENT_IMAGE -o output/gif/${CONTENT_IMAGE%????}__${i%????}.gif
+    cp $DEST_IMAGE output/final/
+done
+mv $CONTENT_INCOMING_DIR/$CONTENT_IMAGE $CONTENT_COMPLETE_DIR/
diff --git a/output/.NOTE.md b/output/.NOTE.md
@@ -0,0 +1 @@
+Output files go here. Including progress pics
diff --git a/output/final/.NOTE.md b/output/final/.NOTE.md
@@ -0,0 +1 @@
+Output files go here. Just final copies
diff --git a/output/gif/.NOTE.md b/output/gif/.NOTE.md
@@ -0,0 +1 @@
+Animated gifs! Hard g
diff --git a/python_requirements.txt b/python_requirements.txt
@@ -0,0 +1,2 @@
+images2gif-Pillow
+Pillow