[Vision] InstanceSegmentation notebook fix: Avoid rendering polygon, …

…if absent (Azure#2364) * Added a check to avoid renderring polygon, if absent * Check if polygon list and contains at least 1 element * Fix black related formating error
vrxmike · Jun 12, 2023 · 1c7d296 · 1c7d296
1 parent f922f0e
commit 1c7d296
Show file tree

Hide file tree

Showing 2 changed files with 74 additions and 33 deletions.
diff --git a/...segmentation-task-fridge-items/automl-image-instance-segmentation-task-fridge-items.ipynb b/...segmentation-task-fridge-items/automl-image-instance-segmentation-task-fridge-items.ipynb
@@ -1,6 +1,7 @@
 {
  "cells": [
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -26,6 +27,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -61,6 +63,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -102,6 +105,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -115,6 +119,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -163,6 +168,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -182,6 +188,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -220,6 +227,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -329,6 +337,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -471,6 +480,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -499,6 +509,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -535,6 +546,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -544,6 +556,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -617,6 +630,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -648,6 +662,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -715,6 +730,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -831,6 +847,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -848,6 +865,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -856,6 +874,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -872,6 +891,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -920,6 +940,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -955,6 +976,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -979,6 +1001,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -998,6 +1021,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1049,13 +1073,15 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
     "# 6. Register best model and deploy"
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1114,6 +1140,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1156,6 +1183,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1212,6 +1240,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1235,6 +1264,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1294,6 +1324,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1360,25 +1391,27 @@
     "        ax.add_patch(rect)\n",
     "        plt.text(topleft_x, topleft_y - 10, label, color=color, fontsize=20)\n",
     "\n",
-    "        polygon_np = np.array(polygon[0])\n",
-    "        polygon_np = polygon_np.reshape(-1, 2)\n",
-    "        polygon_np[:, 0] *= x\n",
-    "        polygon_np[:, 1] *= y\n",
-    "        poly = patches.Polygon(polygon_np, True, facecolor=color, alpha=0.4)\n",
-    "        ax.add_patch(poly)\n",
-    "        poly_line = Line2D(\n",
-    "            polygon_np[:, 0],\n",
-    "            polygon_np[:, 1],\n",
-    "            linewidth=2,\n",
-    "            marker=\"o\",\n",
-    "            markersize=8,\n",
-    "            markerfacecolor=color,\n",
-    "        )\n",
-    "        ax.add_line(poly_line)\n",
+    "        if polygon and len(polygon) > 0:\n",
+    "            polygon_np = np.array(polygon[0])\n",
+    "            polygon_np = polygon_np.reshape(-1, 2)\n",
+    "            polygon_np[:, 0] *= x\n",
+    "            polygon_np[:, 1] *= y\n",
+    "            poly = patches.Polygon(polygon_np, True, facecolor=color, alpha=0.4)\n",
+    "            ax.add_patch(poly)\n",
+    "            poly_line = Line2D(\n",
+    "                polygon_np[:, 0],\n",
+    "                polygon_np[:, 1],\n",
+    "                linewidth=2,\n",
+    "                marker=\"o\",\n",
+    "                markersize=8,\n",
+    "                markerfacecolor=color,\n",
+    "            )\n",
+    "            ax.add_line(poly_line)\n",
     "plt.show()"
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1395,6 +1428,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1412,6 +1446,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": [
@@ -1420,6 +1455,7 @@
    ]
   },
   {
+   "attachments": {},
    "cell_type": "markdown",
    "metadata": {},
    "source": []