Updated the model

KD-3 · Jun 25, 2020 · 028eab2 · 028eab2
1 parent 67b96f9
commit 028eab2
Showing 1 changed file with 29 additions and 12 deletions.
diff --git a/BTP_MobileNet_Model.ipynb b/BTP_MobileNet_Model.ipynb
@@ -5,25 +5,14 @@
     "colab": {
       "name": "BTP_MobileNet_Model.ipynb",
       "provenance": [],
-      "authorship_tag": "ABX9TyM1Ahu+Za3/CgOYP6apODU2",
-      "include_colab_link": true
+      "authorship_tag": "ABX9TyOA6casfH36cpL6wCD15M4m"
     },
     "kernelspec": {
       "name": "python3",
       "display_name": "Python 3"
     }
   },
   "cells": [
-    {
-      "cell_type": "markdown",
-      "metadata": {
-        "id": "view-in-github",
-        "colab_type": "text"
-      },
-      "source": [
-        "<a href=\"https://colab.research.google.com/github/KD-3/DeepLearningMovidiusNCS/blob/master/BTP_MobileNet_Model.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
-      ]
-    },
     {
       "cell_type": "code",
       "metadata": {
@@ -125,6 +114,34 @@
       ],
       "execution_count": null,
       "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "metadata": {
+        "id": "ycyNzNeGsHqM",
+        "colab_type": "code",
+        "colab": {}
+      },
+      "source": [
+        "input_shape = (224, 224, 3)\n",
+        "num_classes = 6\n",
+        "conv_size = 32"
+      ],
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "metadata": {
+        "id": "i2LWFO36sUrb",
+        "colab_type": "code",
+        "colab": {}
+      },
+      "source": [
+        "MobileNet_model = MobileNet(include_top=False, input_shape=input_shape)"
+      ],
+      "execution_count": null,
+      "outputs": []
     }
   ]
 }