[ViPPET] Add Human Pose Estimation model

tools/visual-pipeline-and-platform-evaluation-tool/app.py

@@ -739,6 +739,7 @@ def create_interface(title: str = "Visual Pipeline and Platform Evaluation Tool"
             "ResNet-50 TF (INT8)",
             "PaddleOCR (FP32)",
             "Vehicle Attributes Recognition Barrier 0039 (FP16)",
+            "Human Pose Estimation 0001 (FP16)",
         ],
         value="ResNet-50 TF (INT8)",
         elem_id="object_classification_model",

tools/visual-pipeline-and-platform-evaluation-tool/models.sh

@@ -79,7 +79,8 @@ download_public_models() {
 
 download_open_model_zoo() {
     if [ ! -d /output/public/mobilenet-v2-pytorch ] || \
-       [ ! -d /output/intel/vehicle-attributes-recognition-barrier-0039 ]; then
+       [ ! -d /output/intel/vehicle-attributes-recognition-barrier-0039 ] || \
+       [ ! -d /output/intel/human-pose-estimation-0001 ]; then
         VENV_DIR="$HOME/.venv"
         python3 -m venv "$VENV_DIR"
         source "$VENV_DIR/bin/activate"
@@ -120,7 +121,7 @@ with open(json_path, 'w') as f:
         echo "Model mobilenet-v2-pytorch already exists. Skipping download."
     fi
 
-    # TEMPORARY: download vehicle-attributes-recognition-barrier-0039 until the download script supports it
+    # Download vehicle-attributes-recognition-barrier-0039 model
     if [ ! -d /output/intel/vehicle-attributes-recognition-barrier-0039 ]; then
         omz_downloader --name vehicle-attributes-recognition-barrier-0039 --output_dir /output
         omz_converter --name vehicle-attributes-recognition-barrier-0039 --output_dir /output \
@@ -131,6 +132,18 @@ with open(json_path, 'w') as f:
     else
         echo "Model vehicle-attributes-recognition-barrier-0039 already exists. Skipping download."
     fi
+
+    # Download human-pose-estimation-0001 model
+    if [ ! -d /output/intel/human-pose-estimation-0001 ]; then
+        omz_downloader --name human-pose-estimation-0001 --output_dir /output
+        omz_converter --name human-pose-estimation-0001 --output_dir /output \
+            --download_dir /output
+        cp \
+            /opt/intel/dlstreamer/samples/gstreamer/model_proc/intel/human-pose-estimation-0001.json \
+            /output/intel/human-pose-estimation-0001/human-pose-estimation-0001.json
+    else
+        echo "Model human-pose-estimation-0001 already exists. Skipping download."
+    fi
 }
 
 main

tools/visual-pipeline-and-platform-evaluation-tool/pipelines/simplevs/config.yaml

@@ -33,4 +33,5 @@ parameters:
       - "ResNet-50 TF (INT8)"
       - "PaddleOCR (FP32)"
       - "Vehicle Attributes Recognition Barrier 0039 (FP16)"
+      - "Human Pose Estimation 0001 (FP16)"
     classification_model_default: "PaddleOCR (FP32)"

tools/visual-pipeline-and-platform-evaluation-tool/pipelines/smartnvr/config.yaml

@@ -29,4 +29,5 @@ parameters:
       - "EfficientNet B0 (INT8)"
       - "MobileNet V2 PyTorch (FP16)"
       - "ResNet-50 TF (INT8)"
+      - "Human Pose Estimation 0001 (FP16)"
     classification_model_default: "ResNet-50 TF (INT8)"

tools/visual-pipeline-and-platform-evaluation-tool/utils.py

@@ -212,6 +212,13 @@ def prepare_video_and_constants(
             constants["OBJECT_CLASSIFICATION_MODEL_PROC"] = (
                 f"{MODELS_PATH}/intel/vehicle-attributes-recognition-barrier-0039/vehicle-attributes-recognition-barrier-0039.json"
             )
+        case "Human Pose Estimation 0001 (FP16)":
+            constants["OBJECT_CLASSIFICATION_MODEL_PATH"] = (
+                f"{MODELS_PATH}/intel/human-pose-estimation-0001/FP16/human-pose-estimation-0001.xml"
+            )
+            constants["OBJECT_CLASSIFICATION_MODEL_PROC"] = (
+                f"{MODELS_PATH}/intel/human-pose-estimation-0001/human-pose-estimation-0001.json"
+            )
         case _:
             raise ValueError("Unrecognized Object Classification Model")