Add MocapNET

2026-10-15 00:03:06 +00:00 · 2023-12-15 04:22:41 +05:30
parent 8434144ded
commit 65e8934d5e
336 changed files with 100584 additions and 0 deletions
@@ -0,0 +1,67 @@
+#!/bin/bash
+
+DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+cd "$DIR"
+cd ..
+
+red=$(printf "\033[31m")
+green=$(printf "\033[32m")
+yellow=$(printf "\033[33m")
+blue=$(printf "\033[34m")
+magenta=$(printf "\033[35m")
+cyan=$(printf "\033[36m")
+white=$(printf "\033[37m")
+normal=$(printf "\033[m")
+
+SELECTBODY="--selectJoints 21 hip abdomen chest neck head rcollar rshoulder relbow rhand lcollar lshoulder lelbow lhand rhip rknee rfoot lhip lknee lfoot toe3-2.r toe3-2.l"
+SELECTRHAND="--selectJoints 11 rhand rthumb1 rthumb2 rindex1 rindex2 rmid1 rmid2 rring1 rring2 rpinky1 rpinky2"
+SELECTLHAND="--selectJoints 11 lhand lthumb1 lthumb2 lindex1 lindex2 lmid1 lmid2 lring1 lring2 lpinky1 lpinky2"
+
+datasetDir="dataset/MotionCapture"
+datasetSubDir="01 02 03 05 06 07 08 09 10 11 12 13 14 15 16 17 18 19"
+outputDir="dataset/"
+
+rm $outputDir/2d_body_test.csv
+rm $outputDir/3d_body_test.csv
+rm $outputDir/bvh_body_test.csv
+
+#create a small test file with one example  
+#rm $outputDir/testSMALL.csv
+# ./GroundTruthDumper --from MotionCapture/01/01_01.bvh  --offsetPositionRotation 0 750 2500 0 0 0 --maxFrames 1 --csv $outputDir testSMALL.csv 
+ 
+#PRODUCESVG="--svg $outputDir/svg"
+PRODUCESVG=" "
+
+
+datasetsProcessed=0
+totaldatasetsToGo=`echo "$datasetSubDir" | awk '{print NF}'`
+
+rm $outputDir/test.csv
+ for d in $datasetSubDir
+           do  
+             if [ -d $datasetDir/$d ]
+              then
+              echo "$green Found $d directory $normal" 
+              datasetFile=`ls $datasetDir/$d | grep bvh`
+              ((datasetsProcessed++)) 
+              #-------------------------------------------------------
+             
+                for f in $datasetFile
+                  do
+                  echo "$green Generating Test - $datasetsProcessed/$totaldatasetsToGo - $datasetDir/$d/$f file $normal" 
+                    #||||||||||||||||||||||||||||||||||||||||||||||||||||||||
+                       BVHFILE="$datasetDir/$d/$f"                       # --svg $outputDir 
+                       #./GroundTruthDumper --from $BVHFILE  --csv $outputDir test.csv# --bvh $outputDir/$f-random.bvh
+                       ./GroundTruthDumper --from $BVHFILE  $SELECTBODY --offsetPositionRotation 0 750 2000 0 0 0 --occlusions --csv $outputDir body_test.csv 2d+bvh $PRODUCESVG #--to ~/Documents/Programming/RGBDAcquisition/opengl_acquisition_shared_library/opengl_depth_and_color_renderer/Scenes/testBVH.conf --bvh $outputDir/$f-random.bvh --svg $outputDir/svg
+                       #exit 0
+                       #./GroundTruthDumper --from $BVHFILE --setPositionRotation 0 400 0 0 0 0  --csv $outputDir test.csv# --bvh $outputDir/$f-random.bvh
+                    #|||||||||||||||||||||||||||||||||||||||||||||||||||||||| 
+                done
+
+              #-------------------------------------------------------
+             else
+              echo "$red Could not find $d directory $normal"
+             fi
+  done 
+  
+exit 0