downloading model via notebook utils not omz

Author: Roszczyk

notebooks/406-3D-pose-estimation-webcam/406-3D-pose-estimation.ipynb

@@ -49,7 +49,14 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%pip install pythreejs \"openvino-dev>=2023.1.0\""
+    "%pip install -q pythreejs \"openvino-dev>=2023.1.0\"\n",
+    "\n",
+    "# Fetch `notebook_utils` module\n",
+    "import urllib.request\n",
+    "urllib.request.urlretrieve(\n",
+    "    url='https://raw.githubusercontent.com/openvinotoolkit/openvino_notebooks/main/notebooks/utils/notebook_utils.py',\n",
+    "    filename='notebook_utils.py'\n",
+    ")"
    ]
   },
   {
@@ -63,7 +70,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "id": "316ad889-8514-430f-baf4-4f32abd43356",
    "metadata": {},
    "outputs": [],
@@ -72,14 +79,19 @@
     "import sys\n",
     "import time\n",
     "from pathlib import Path\n",
+    "import tarfile\n",
     "\n",
     "import cv2\n",
     "import ipywidgets as widgets\n",
     "import numpy as np\n",
     "from IPython.display import clear_output, display\n",
     "import openvino as ov\n",
     "\n",
-    "sys.path.append(\"../utils\")\n",
+    "import torch\n",
+    "import torch.onnx\n",
+    "import torchvision.models as models\n",
+    "from torch.autograd import Variable\n",
+    "\n",
     "import notebook_utils as utils\n",
     "\n",
     "sys.path.append(\"./engine\")\n",
@@ -98,7 +110,7 @@
     "### Download the model\n",
     "[back to top ⬆️](#Table-of-contents:)\n",
     "\n",
-    "We use `omz_downloader`, which is a command line tool from the `openvino-dev` package. `omz_downloader` automatically creates a directory structure and downloads the selected model."
+    "Use the `download_file`, a function from the `notebook_utils` file. It automatically creates a directory structure and downloads the selected model. This step is skipped if the package is already downloaded and unpacked. The chosen model comes from the public directory, which means it must be converted into OpenVINO Intermediate Representation (OpenVINO IR)."
    ]
   },
   {
@@ -109,25 +121,32 @@
    "outputs": [],
    "source": [
     "# directory where model will be downloaded\n",
-    "base_model_dir = \"model\"\n",
+    "base_model_dir = Path(\"model\")\n",
     "\n",
     "# model name as named in Open Model Zoo\n",
-    "model_name = \"human-pose-estimation-3d-0001\"\n",
+    "short_model_name = \"human-pose-estimation-3d\"\n",
+    "model_name = f\"{short_model_name}-0001\"\n",
     "# selected precision (FP32, FP16)\n",
     "precision = \"FP32\"\n",
     "\n",
-    "BASE_MODEL_NAME = f\"{base_model_dir}/public/{model_name}/{model_name}\"\n",
+    "MAIN_MODEL_PATH=f\"{base_model_dir}/public/{model_name}\"\n",
+    "BASE_MODEL_NAME = f\"{MAIN_MODEL_PATH}/{model_name}\"\n",
     "model_path = Path(BASE_MODEL_NAME).with_suffix(\".pth\")\n",
     "onnx_path = Path(BASE_MODEL_NAME).with_suffix(\".onnx\")\n",
     "\n",
-    "ir_model_path = f\"model/public/{model_name}/{precision}/{model_name}.xml\"\n",
-    "model_weights_path = f\"model/public/{model_name}/{precision}/{model_name}.bin\"\n",
+    "archive_name=Path(f\"{short_model_name}.tar.gz\")\n",
+    "model_url = f\"https://storage.openvinotoolkit.org/repositories/open_model_zoo/public/2022.1/{model_name}/{archive_name}\"\n",
+    "\n",
+    "downloaded_model_path= Path(MAIN_MODEL_PATH) / archive_name\n",
+    "if not downloaded_model_path.exists():\n",
+    "    utils.download_file(model_url, downloaded_model_path.name, downloaded_model_path.parent)\n",
+    "\n",
+    "ir_model_path = Path(f\"model/public/{model_name}/{precision}/{model_name}.xml\")\n",
+    "model_weights_path = Path(f\"model/public/{model_name}/{precision}/{model_name}.bin\")\n",
     "\n",
     "if not model_path.exists():\n",
-    "    download_command = (\n",
-    "        f\"omz_downloader \" f\"--name {model_name} \" f\"--output_dir {base_model_dir}\"\n",
-    "    )\n",
-    "    ! $download_command"
+    "    with tarfile.open(downloaded_model_path) as file:\n",
+    "        file.extractall(MAIN_MODEL_PATH)"
    ]
   },
   {
@@ -145,7 +164,9 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "c9bdfdee-c2ef-4710-96c1-8a6a896a8cba",
-   "metadata": {},
+   "metadata": {
+    "scrolled": true
+   },
    "outputs": [],
    "source": [
     "if not onnx_path.exists():\n",
@@ -204,7 +225,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 6,
    "id": "92a04102-aebf-4976-874b-b98dca97ec48",
    "metadata": {},
    "outputs": [],
@@ -257,7 +278,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 8,
    "id": "08f8055b-a6cf-4003-8232-6f73a86d6034",
    "metadata": {},
    "outputs": [],
@@ -306,7 +327,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 9,
    "id": "22fd3e08-ed3b-44ac-bd07-4a80130d6681",
    "metadata": {},
    "outputs": [],
@@ -394,7 +415,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 10,
    "id": "3be526d0-75ad-4bd1-85b1-ca8185eca918",
    "metadata": {
     "tags": []
@@ -587,7 +608,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 11,
    "id": "3f82e298-5912-48c7-90b5-339aea3c177d",
    "metadata": {
     "tags": []
@@ -603,6 +624,14 @@
     "\n",
     "run_pose_estimation(source=source, flip=isinstance(source, int), use_popup=False)"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e3c77b7c-5211-4715-ad48-4ec38a82941c",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {
@@ -621,20 +650,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.10"
-  },
-  "openvino_notebooks": {
-   "imageUrl": "https://github.com/openvinotoolkit/openvino_notebooks/blob/main/notebooks/406-3D-pose-estimation-webcam/406-3D-pose-estimation.gif?raw=true",
-   "tags": {
-    "categories": [
-     "Live Demos"
-    ],
-    "libraries": [],
-    "other": [],
-    "tasks": [
-     "Pose Estimation"
-    ]
-   }
+   "version": "3.11.5"
   },
   "widgets": {
    "application/vnd.jupyter.widget-state+json": {