Add MocapNET

animation/MocapNET/scripts/dump_and_process_video.sh

@@ -0,0 +1,93 @@
+#!/bin/bash 
+
+
+DATASET=""
+if (( $#<1 ))
+then 
+ echo "Please provide arguments first argument is dataset "
+ exit 1
+else
+ DATASET=$1-data
+fi
+# $1 holds the video file we want to process ( it should be something like path/to/videofile.mp4 )
+#DATASET now holds the output directory we will create ( it should be something like path/to/videofile.mp4-data/ )
+
+
+#Remember the directory where user started
+STARTDIR=`pwd`
+
+#Remember the directory where the script is ( and where MocapNET is :) )
+DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+
+#Please give the path to openpose here..
+OPENPOSE_DIR="/home/ammar/Documents/3dParty/openpose/"
+OPENPOSE_BINARY_DIR="/home/ammar/Documents/3dParty/openpose/build/examples/openpose/"
+ 
+#Dataset dumping using ffmpeg
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+#Create the new dataset directory
+mkdir $DATASET
+
+#Dump the video files using our naming scheme
+ffmpeg -i $1  -r 30 -q:v 1  $DATASET/colorFrame_0_%05d.jpg
+
+#Make sure we start at frame 0
+cp $DATASET/colorFrame_0_00001.jpg $DATASET/colorFrame_0_00000.jpg
+
+#We now want to grab an absolute path to our dataset
+cd $DATASET
+FULL_PATH_TO_DATASET=`pwd`
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+
+
+
+#2D pose estimation using OpenPose
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+cd $OPENPOSE_DIR
+LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64 $OPENPOSE_BINARY_DIR/openpose.bin -number_people_max 1 --hand --face --write_json $FULL_PATH_TO_DATASET  -image_dir $FULL_PATH_TO_DATASET  $@
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+
+
+
+#2D pose conversion to CSV using our tool
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+cd "$DIR"
+cd ..
+
+./convertOpenPoseJSONToCSV  --from $DATASET
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+
+
+
+#3D pose estimation using MocapNET2
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+cd "$DIR"
+cd ..
+
+./MocapNET2CSV --from $FULL_PATH_TO_DATASET/2dJoints_v1.4.csv --mt --show 3 --save
+cp out.bvh $FULL_PATH_TO_DATASET/predicted.bvh
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+#-----------------------------------------------------------------------------
+
+
+
+
+
+cd $STARTDIR 
+exit 0
+