1 files changed, 99 insertions, 0 deletions
diff --git a/warp-mix.sh b/warp-mix.sh
new file mode 100644
index 0000000..2f07b2b
--- /dev/null
+++ b/warp-mix.sh
@@ -0,0 +1,99 @@
+#!/bin/bash
+
+BASENAME=tmp
+ITER=1
+FRAMES=200
+A_DIR="A"
+B_DIR="B"
+POSITIONAL=()
+while [[ $# -gt 0 ]]
+do
+key="$1"
+
+case $key in
+    -f|--frames)
+    FRAMES=$(expr $2 + 0)
+    shift # past argument
+    shift # past value
+    ;;
+    -s|--start)
+    ITER=$(expr $2 + 1)
+    shift # past argument
+    shift # past value
+    ;;
+    -o|--output)
+    OUTPUT_DIR="$2"
+    BASENAME=$(basename $OUTPUT_DIR)
+    shift # past argument
+    shift # past value
+    ;;
+    -a|--a_dir)
+    A_DIR="$2"
+    shift # past argument
+    shift # past value
+    ;;
+    -b|--b_dir)
+    B_DIR="$2"
+    shift # past argument
+    shift # past value
+    ;;
+    # --default)
+    # DEFAULT=YES
+    # shift # past argument
+    # ;;
+    *)    # unknown option
+    POSITIONAL+=("$1") # save it in an array for later
+    shift # past argument
+    ;;
+esac
+done
+set -- "${POSITIONAL[@]}" # restore positional parameters
+
+rm -rf tmp/
+mkdir tmp/
+
+echo $dirs
+
+a=$(printf "%s/frame_%05d.png" $A_DIR $ITER)
+b=$(printf "%s/frame_%05d.png" $B_DIR $((ITER + FRAMES))
+
+convert -verbose "$DIRECTORY/$a" PNG24:tmp_normalized_0.png
+convert -verbose "$DIRECTORY/$b" PNG24:tmp_normalized_1.png
+
+./makeOptFlow.sh tmp_normalized_%01d.png flow 0
+
+STEP=$(bc -l <<< "1 / $FRAMES")
+SEQUENCE=`seq 0.0 $STEP 1.0`
+
+for scale in $SEQUENCE; do
+  scale="${scale/,/.}"
+  echo "${ITER}: ${scale}"
+  a_f=$(printf "%s/frame_%05d.png" $A_DIR $ITER)
+  b_f=$(printf "%s/frame_%05d.png" $B_DIR $ITER)
+  th torch_warp.lua \
+    -flow_file flow/backward_1_0.flo \
+    -source_image $a_f \
+    -output_image warp_0.png \
+    -scale $scale
+  th torch_warp.lua \
+    -flow_file flow/forward_0_1.flo \
+    -source_image $b_f \
+    -output_image warp_1.png \
+    -scale $(bc <<< "1.0-$scale")
+  OUT=$(printf "tmp/frame_%05d.png" $ITER)
+  convert warp_0.png warp_1.png -compose blend -define compose:args=$(bc <<< "100*$scale/1") -composite $OUT
+  rm warp_0.png warp_1.png
+
+  ITER=$((ITER+1))
+done
+
+rm -rf flow/ tmp_normalized*.png
+
+ffmpeg -i tmp/frame_%05d.png -vcodec libx264 -pix_fmt yuv420p -profile:v baseline -level 3 "mixed_${BS}.mp4"
+
+cd tmp/
+mkdir 256
+for i in *; do convert $i -resize '256x256!' 256/$i; done
+cd ..
+mv tmp mixed_${BS}
+