stylize.sh


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

#!/bin/bash

set -e ### exit the script if any part of it fails

cwd=$(pwd | sed -r 's%.*/%%g')
if [ $cwd != "neural-art" ] ; then
    echo "You must be in the 'neural-art' directory when you run this"
    exit 1
fi

if [ $# -eq 0 ] ; then
    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 as well"
    echo "Example: ./stylize.sh Images/Jitter_Doll.jpg Images/cute.jpg"
    exit 1
fi

./clear_dir.sh # of now useless files

# stylize data [pair (style, content)]
python neuralart.py $1 $2

# render images (actual frames) from (an) images.npy
python renderer.py

# fix render artifacts
python renderer.py --fix

# wrap 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" $(basename ${2%.*})'_in_'$(basename ${1%.*})'.mp4'