Added openpose installation

pages/Pose_And_Motion_Based_SVM_Model.py

@@ -1,16 +1,83 @@
+#pip install pyopenpose
+
+#===================IMPORT LIBRARIES==================================
 import cv2
-import pyopenpose as op
+#import pyopenpose as op
 import numpy as np
 import joblib
 import tempfile
 import streamlit as st
 import subprocess
+import os
 from os.path import basename, splitext, exists
 from sklearn.preprocessing import StandardScaler
 
+
 # Streamlit app header
 st.title("Pose-Based One-Class SVM Model")
 
+#==========================INSTALLATIONS=====================================
+# Install OpenPose
+def install_openpose():
+    git_repo_url = 'https://github.com/CMU-Perceptual-Computing-Lab/openpose.git'
+    project_name = splitext(basename(git_repo_url))[0]
+
+    # Remove any existing openpose folder
+    if os.path.exists("openpose"):
+        subprocess.run(["rm", "-rf", "openpose"])
+
+    # Install CMake 3.17 for CUDA10
+    if not exists(project_name):
+        st.write("Installing CMake and OpenPose dependencies...")
+
+        subprocess.run(["wget", "-q", "https://cmake.org/files/v3.17/cmake-3.17.2-Linux-x86_64.tar.gz"])
+        subprocess.run(["tar", "xfz", "cmake-3.17.2-Linux-x86_64.tar.gz", "--strip-components=1", "-C", "/usr/local"])
+
+        # Clone OpenPose repository
+        subprocess.run(["git", "clone", "--depth", "1", git_repo_url])
+
+        # Update the CMakeLists.txt for compatibility
+        subprocess.run([
+            "sed", "-i",
+            r"s/execute_process(COMMAND git checkout --recursive master WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}\/3rdparty\/caffe)/execute_process(COMMAND git checkout f019d0dfe86f49d1140961f8c7dec22130c83154 WORKING_DIRECTORY ${CMAKE_SOURCE_DIR}\/3rdparty\/caffe)/g",
+            "openpose/CMakeLists.txt"
+        ])
+
+        subprocess.run(["cd", "openpose", "&&", "git", "submodule", "update", "--init", "--recursive", "--remote"])
+
+        # Install system dependencies
+        dependencies = [
+            "libatlas-base-dev", "libprotobuf-dev", "libleveldb-dev", "libsnappy-dev",
+            "libhdf5-serial-dev", "protobuf-compiler", "libgflags-dev",
+            "libgoogle-glog-dev", "liblmdb-dev", "opencl-headers", "ocl-icd-opencl-dev",
+            "libviennacl-dev"
+        ]
+        subprocess.run(["apt-get", "-qq", "install", "-y"] + dependencies)
+
+        # Build OpenPose
+        os.makedirs("openpose/build", exist_ok=True)
+        subprocess.run(["cmake", ".."], cwd="openpose/build")
+        subprocess.run(["make", "-j", str(os.cpu_count())], cwd="openpose/build")
+
+        st.write("OpenPose installation complete.")
+
+# Call the installation function
+install_openpose()
+
+# Copy the models.zip file to the correct directory and unzip
+def copy_and_unzip_models():
+    if os.path.exists("/content/drive/MyDrive/models.zip"):
+        subprocess.run(["cp", "/content/drive/MyDrive/models.zip", "/content/"])
+        subprocess.run(["unzip", "-o", "models.zip", "-d", "openpose"])
+
+        st.write("Models copied and unzipped successfully.")
+    else:
+        st.write("Model file not found. Ensure it’s located in /content/drive/MyDrive/models.zip")
+
+# Call the function to copy and unzip models
+copy_and_unzip_models()
+
+
 #==========================IMPORTING MODELS==========================
 pose_scaler = joblib.load('pages/pose_scaler.pkl')  # Standard scaler for pose data
 ocsvm_pose = joblib.load('pages/ocsvm_pose_model-4.pkl')  # Pre-trained One-Class SVM for pose