Update transform example

nekitmm · Aug 17, 2020 · 2c778fc · 2c778fc
1 parent aedf0b1
commit 2c778fc
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/starnet_v1_TF2_transform.ipynb b/starnet_v1_TF2_transform.ipynb
@@ -13,6 +13,13 @@
     "from starnet_v1_TF2 import StarNet"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<p>Load a model. There are two modes - RGB and Greyscale.</p>"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 12,
@@ -23,6 +30,13 @@
     "starnet.load_model('./weights', './history')"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<p>Load and transform an image.</p>"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 13,