Add entry-level function

isic-superpixel.ipynb

@@ -273,6 +273,37 @@
     "        return superpixel_labels"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6aab4ef1-49f4-4146-a86b-7806ce47b8b5",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def superpixels(filename, variant=1) -> np.ndarray:\n",
+    "    \"\"\"\n",
+    "    Segment an image in the file into super-pixels. Variants:\n",
+    "    1=old implementation\n",
+    "    2=old implementation with bigger superpixels\n",
+    "    3=new implementation (with bigger superpixels) using texture\n",
+    "    4=new implementation (with bigger superpixels) using texture with scikit-image SLIC\n",
+    "    \"\"\"\n",
+    "    if variant == 1:\n",
+    "        image = ISICSuperpixelV2.load_image(filename)\n",
+    "        return ISICSuperpixelV2._slic(image, num_segments=1000)\n",
+    "    elif variant == 2:\n",
+    "        image = ISICSuperpixelV2.load_image(filename)\n",
+    "        return ISICSuperpixelV2._slic(image, num_segments=300)\n",
+    "    elif variant == 3:\n",
+    "        image = itk.imread(filename)\n",
+    "        return ISICSuperpixelV3._slic(image, num_segments=300, slic_zero=False)\n",
+    "    elif variant == 4:\n",
+    "        image = itk.imread(filename)\n",
+    "        return ISICSuperpixelV3._slic(image, num_segments=300, slic_zero=True)\n",
+    "    else:\n",
+    "        raise ValueError"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -288,7 +319,8 @@
     "image_file = image_list[0]\n",
     "image_file = \"./images/ISIC_0016085.JPG\"\n",
     "image = ISICSuperpixelV2.load_image(image_file)\n",
-    "superpixel_labels = ISICSuperpixelV2._slic(image, num_segments=300)"
+    "superpixel_labels = ISICSuperpixelV2._slic(image, num_segments=300)\n",
+    "superpixel_labels = superpixels(image_file, 4)"
    ]
   },
   {