Notebooks minor updates for support testing (#1113)

* Notebooks minor updates for support testing
sony · Jun 19, 2024 · c280efa · c280efa
1 parent d498eac
commit c280efa
Show file tree

Hide file tree

Showing 4 changed files with 46 additions and 17 deletions.
diff --git a/tutorials/notebooks/imx500_notebooks/keras/example_keras_effdet_lite0_for_imx500.ipynb b/tutorials/notebooks/imx500_notebooks/keras/example_keras_effdet_lite0_for_imx500.ipynb
@@ -94,6 +94,7 @@
    "source": [
     "import sys\n",
     "import importlib\n",
+    "import os\n",
     "\n",
     "if not importlib.util.find_spec('model_compression_toolkit'):\n",
     "    !pip install model_compression_toolkit\n",
@@ -134,12 +135,13 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "!wget -nc http://images.cocodataset.org/annotations/annotations_trainval2017.zip\n",
-    "!unzip -q -o annotations_trainval2017.zip -d ./coco\n",
-    "!echo Done loading annotations\n",
-    "!wget -nc http://images.cocodataset.org/zips/val2017.zip\n",
-    "!unzip -q -o val2017.zip -d ./coco\n",
-    "!echo Done loading val2017 images"
+    "if not os.path.isdir('coco'):\n",
+    "    !wget -nc http://images.cocodataset.org/annotations/annotations_trainval2017.zip\n",
+    "    !unzip -q -o annotations_trainval2017.zip -d ./coco\n",
+    "    !echo Done loading annotations\n",
+    "    !wget -nc http://images.cocodataset.org/zips/val2017.zip\n",
+    "    !unzip -q -o val2017.zip -d ./coco\n",
+    "    !echo Done loading val2017 images"
    ],
    "metadata": {
     "collapsed": false

diff --git a/tutorials/notebooks/imx500_notebooks/keras/keras_deeplabv3plus_for_imx500.ipynb b/tutorials/notebooks/imx500_notebooks/keras/keras_deeplabv3plus_for_imx500.ipynb
@@ -65,9 +65,12 @@
    "execution_count": null,
    "outputs": [],
    "source": [
-    "!wget -nc http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar\n",
-    "!tar -xf VOCtrainval_11-May-2012.tar\n",
-    "!echo Done loading PascelVOC 2012"
+    "import os\n",
+    "\n",
+    "if not os.path.isdir('VOCdevkit'):\n",
+    "    !wget -nc http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar\n",
+    "    !tar -xf VOCtrainval_11-May-2012.tar\n",
+    "    !echo Done loading PascelVOC 2012"
    ],
    "metadata": {
     "collapsed": false

diff --git a/tutorials/notebooks/imx500_notebooks/pytorch/pytorch_yolov8n_pose_for_imx500.ipynb b/tutorials/notebooks/imx500_notebooks/pytorch/pytorch_yolov8n_pose_for_imx500.ipynb
@@ -286,7 +286,10 @@
     "representative_dataset_gen = DataLoader(representative_dataset, BATCH_SIZE, shuffle=True)"
    ],
    "metadata": {
-    "collapsed": false
+    "collapsed": false,
+    "tags": [
+     "long_run"
+    ]
    },
    "id": "20fe96b6cc95d38c"
   },
@@ -321,7 +324,10 @@
     "print('Quantized-GPTQ model is ready')"
    ],
    "metadata": {
-    "collapsed": false
+    "collapsed": false,
+    "tags": [
+     "long_run"
+    ]
    },
    "id": "240421e00f6cce34"
   },
@@ -347,7 +353,10 @@
     "                                  repr_dataset=representative_dataset_gen)"
    ],
    "metadata": {
-    "collapsed": false
+    "collapsed": false,
+    "tags": [
+     "long_run"
+    ]
    },
    "id": "546ff946af81702b"
   },
@@ -457,7 +466,10 @@
     "print(\"Quantized using GPTQ model mAP: {:.4f}\".format(eval_results[0]))"
    ],
    "metadata": {
-    "collapsed": false
+    "collapsed": false,
+    "tags": [
+     "long_run"
+    ]
    },
    "id": "168468f17ae8bc59"
   },

diff --git a/tutorials/notebooks/imx500_notebooks/pytorch/pytorch_yolov8n_seg_for_imx500.ipynb b/tutorials/notebooks/imx500_notebooks/pytorch/pytorch_yolov8n_seg_for_imx500.ipynb
@@ -300,7 +300,10 @@
      "base_uri": "https://localhost:8080/"
     },
     "id": "20fe96b6cc95d38c",
-    "outputId": "22b0be33-ef7b-490a-82ae-7eb02a3474a2"
+    "outputId": "22b0be33-ef7b-490a-82ae-7eb02a3474a2",
+    "tags": [
+     "long_run"
+    ]
    },
    "outputs": [],
    "source": [
@@ -340,7 +343,10 @@
    "execution_count": null,
    "id": "240421e00f6cce34",
    "metadata": {
-    "id": "240421e00f6cce34"
+    "id": "240421e00f6cce34",
+    "tags": [
+     "long_run"
+    ]
    },
    "outputs": [],
    "source": [
@@ -380,7 +386,10 @@
      "base_uri": "https://localhost:8080/"
     },
     "id": "546ff946af81702b",
-    "outputId": "cf627960-7b8b-423c-8cae-fbccddcb76f3"
+    "outputId": "cf627960-7b8b-423c-8cae-fbccddcb76f3",
+    "tags": [
+     "long_run"
+    ]
    },
    "outputs": [],
    "source": [
@@ -463,7 +472,10 @@
    "execution_count": null,
    "id": "VLwCbC2_szpy",
    "metadata": {
-    "id": "VLwCbC2_szpy"
+    "id": "VLwCbC2_szpy",
+    "tags": [
+     "long_run"
+    ]
    },
    "outputs": [],
    "source": [