Add MocapNET

animation/MocapNET/src/python/mnet4/setup.sh

@@ -0,0 +1,124 @@
+#!/bin/bash
+DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+cd "$DIR"
+
+#Simple dependency checker that will apt-get stuff if something is missing
+# sudo apt-get install python3-venv python3-pip
+SYSTEM_DEPENDENCIES="build-essential python3-venv python3-pip freeglut3-dev libglew-dev protobuf-compiler libprotobuf-dev libavdevice-dev"
+
+for REQUIRED_PKG in $SYSTEM_DEPENDENCIES
+do
+PKG_OK=$(dpkg-query -W --showformat='${Status}\n' $REQUIRED_PKG|grep "install ok installed")
+echo "Checking for $REQUIRED_PKG: $PKG_OK"
+if [ "" = "$PKG_OK" ]; then
+
+  echo "No $REQUIRED_PKG. Setting up $REQUIRED_PKG."
+
+  #If this is uncommented then only packages that are missing will get prompted..
+  #sudo apt-get --yes install $REQUIRED_PKG
+
+  #if this is uncommented then if one package is missing then all missing packages are immediately installed..
+  sudo apt-get --yes install $SYSTEM_DEPENDENCIES  
+  break
+fi
+done
+#------------------------------------------------------------------------------
+
+HEADLESS=" --headless"
+OPENCV=" "
+
+if [[ $* == *--collab* ]]
+then 
+ echo "Collab mode!"
+ echo "Performing python package installation without a virtual environment"
+HEADLESS=" --headless"
+OPENCV="opencv-python-headless"
+rm -rf /content/sample_data/ #remove sample data
+ln -s $DIR /content/mnet4
+sudo apt-get --yes install build-essential freeglut3-dev libglew-dev #For some reason collab has some errors on previous check so just double-make sure we have these libraries
+
+else 
+ echo "Using a python pythonVirtualEnvironment environment"
+ echo "Activate it using : source pythonVirtualEnvironment/bin/activate"
+ python3 -m venv pythonVirtualEnvironment
+ source pythonVirtualEnvironment/bin/activate
+OPENCV="opencv-python"
+HEADLESS=" "
+fi
+
+
+
+cd "$DIR"
+if [ -f RGBDAcquisition/README.md ]; then
+echo "RGBDAcquisition appears to already exist .."
+else
+ git clone https://github.com/AmmarkoV/RGBDAcquisition
+ cd RGBDAcquisition
+ #----------------------------------------------------------------------------------------------------------------------
+ #mkdir build
+ #cd build
+ #cmake ..
+ #cd ..
+ #----------------------------------------------------------------------------------------------------------------------
+ cd opengl_acquisition_shared_library/opengl_depth_and_color_renderer/
+ mkdir build 
+ cd build
+ cmake ..
+ make BVHTester
+ #----------------------------------------------------------------------------------------------------------------------
+ cd "$DIR"
+ ln -s RGBDAcquisition/opengl_acquisition_shared_library/opengl_depth_and_color_renderer/src/Applications/BVHTester BVH
+ ln -s RGBDAcquisition/opengl_acquisition_shared_library/opengl_depth_and_color_renderer/BVHTester GroundTruthDumper
+fi
+
+
+ 
+ python3 -m pip install --upgrade pip #2.8.0 nvidia-tensorrt
+ python3 -m pip install tensorflow numpy numba tensorboard_plugin_profile tensorflow-model-optimization keras pillow tf2onnx onnxruntime onnx matplotlib pydot mediapipe $OPENCV scikit-learn
+
+ 
+ #Two ways to get Models!
+ #One through the database and compile them yourself (commented out to save Google Collab CPU time)..
+ #python3 -m getModelFromDatabase --get 318
+ #To generate the archive in the next lines do :
+ #   zip mnet4_318.zip step1_*/*
+
+ #Another just take the precompiled ensembles ready to run
+ wget http://ammar.gr/mocapnet/mnet4/mnet4_-314.zip -O mnet4_-314.zip
+ unzip -o mnet4_-314.zip #<- upgrade to 324 mnet4_318.zip
+
+ #Unzip models for PoseNET.py operation..
+ wget http://ammar.gr/mocapnet/mnet4/mediapipe.zip -O mediapipe.zip
+ unzip -o mediapipe.zip
+
+ #To get :
+ # python3 -m evaluateMocapNET --config dataset/body_configuration.json --all body --engine onnx 
+ #  to work
+
+ #wget http://cvrlcode.ics.forth.gr/web_share/mocapnet/MocapNETBodyValidationSet.zip
+ #wget http://ammar.gr/datasets/MocapNETBodyValidationSet.zip
+ #unzip MocapNETBodyValidationSet.zip
+
+
+
+ #Doing a test run!
+
+# echo "To enable Tensorflow profiling for non-root users please add : "
+# echo "options nvidia \"NVreg_RestrictProfilingToAdminUsers=0\""
+# echo "to /etc/modprobe.d/nvidia-graphics-drivers.conf"
+# echo " "
+ echo "Ready for use"
+ echo "From now on to work with MocapNET you can use :  "
+ echo "source pythonVirtualEnvironment/bin/activate"
+ echo "!python3 -m mediapipeHolisticWebcamMocapNET --from webcam  --save --plot"
+ echo " OR "
+ echo "!python3 -m mediapipeHolisticWebcamMocapNET --from shuffle.webm --save --plot"
+ echo " OR "
+ echo "!python3 -m csvNET --from ../../../dataset/sample.csv  --save --plot"
+ echo "  "
+ echo " if you are running on collab don't forget to append --headless to your command"
+ echo " after the run is complete out.bvh will contain your results  "
+
+#Force making of library during last-stage of setup..
+BVH/makeLibrary.sh
+