Add files/code from https://github.com/manuelruder/artistic-videos

1 files changed, 56 insertions, 0 deletions

diff --git a/makeOptFlow.sh b/makeOptFlow.sh
new file mode 100755
index 0000000..0edbb77
--- /dev/null
+++ b/makeOptFlow.sh
@@ -0,0 +1,56 @@
+# Specify the path to the optical flow utility here.
+# Also check line 44 and 47 whether the arguments are in the correct order.
+flowCommandLine="bash run-deepflow.sh"
+
+if [ -z "$flowCommandLine" ]; then
+  echo "Please open makeOptFlow.sh and specify the command line for computing the optical flow."
+  exit 1
+fi
+
+if [ ! -f ./consistencyChecker/consistencyChecker ]; then
+  if [ ! -f ./consistencyChecker/Makefile ]; then
+    echo "Consistency checker makefile not found."
+    exit 1
+  fi
+  cd consistencyChecker/
+  make
+  cd ..
+fi
+
+filePattern=$1
+folderName=$2
+startFrame=${3:-1}
+stepSize=${4:-1}
+
+if [ "$#" -le 1 ]; then
+   echo "Usage: ./makeOptFlow <filePattern> <outputFolder> [<startNumber> [<stepSize>]]"
+   echo -e "\tfilePattern:\tFilename pattern of the frames of the videos."
+   echo -e "\toutputFolder:\tOutput folder."
+   echo -e "\tstartNumber:\tThe index of the first frame. Default: 1"
+   echo -e "\tstepSize:\tThe step size to create long-term flow. Default: 1"
+   exit 1
+fi
+
+i=$[$startFrame]
+j=$[$startFrame + $stepSize]
+
+mkdir -p "${folderName}"
+
+while true; do
+  file1=$(printf "$filePattern" "$i")
+  file2=$(printf "$filePattern" "$j")
+  if [ -a $file2 ]; then
+    if [ ! -f ${folderName}/forward_${i}_${j}.flo ]; then
+      eval $flowCommandLine "$file1" "$file2" "${folderName}/forward_${i}_${j}.flo"
+    fi
+    if [ ! -f ${folderName}/backward_${j}_${i}.flo ]; then
+      eval $flowCommandLine "$file2" "$file1" "${folderName}/backward_${j}_${i}.flo"
+    fi
+    ./consistencyChecker/consistencyChecker "${folderName}/backward_${j}_${i}.flo" "${folderName}/forward_${i}_${j}.flo" "${folderName}/reliable_${j}_${i}.pgm"
+    ./consistencyChecker/consistencyChecker "${folderName}/forward_${i}_${j}.flo" "${folderName}/backward_${j}_${i}.flo" "${folderName}/reliable_${i}_${j}.pgm"
+  else
+    break
+  fi
+  i=$[$i +1]
+  j=$[$j +1]
+done