mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-11 21:41:18 +00:00
Justin John
94 lines
3.5 KiB
Bash
94 lines
3.5 KiB
Bash
#!/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
|
|
|