Minor modifications

Author: Anastasia

Keras-Transfer-Learning/transfer-learning-vgg.ipynb

@@ -38,6 +38,20 @@
     "os.environ['TF_DETERMINISTIC_OPS'] = '1'"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<h2>Experiment 1: run model without fully-connected layers</h2>"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<h3>1. Initialize the model</h3>"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -121,6 +135,13 @@
     "vgg_conv.summary()"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<h3>2. Prepare the training data</h3>"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -180,7 +201,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "Let's explore the train dataset:"
+    "<h3>3. Visualize the training dataset</h3>"
    ]
   },
   {
@@ -237,6 +258,13 @@
     "plt.title(\"Class name: {}\".format(classes_list[train_image_label_id]))"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<h3>4. Obtain model predictions on the train data</h3>"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -272,6 +300,13 @@
     "print(\"Train features: {}\".format(train_features_vec.shape))"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<h3>5. Prepare the validation data</h3>"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 9,
@@ -298,6 +333,13 @@
     "    shuffle=False)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<h3>6. Visualize the validation dataset</h3>"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 10,
@@ -352,6 +394,13 @@
     "plt.title(\"Class name: {}\".format(classes_list[val_image_label_id]))"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<h3>7. Obtain model predictions on the validation data</h3>"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 11,
@@ -383,7 +432,14 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "Create your own model and train the network:"
+    "<h2>Experiment 2: Create your own model and train the network</h2>"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<h3>1. Configure and train the model</h3>"
    ]
   },
   {
@@ -458,6 +514,13 @@
     "                    validation_data=(validation_features_vec,validation_labels))"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<h3>2. Show the errors</h3>"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 13,
@@ -514,13 +577,6 @@
     "print(\"Number of errors = {}/{}\".format(len(errors),nVal))"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "metadata": {},
-   "source": [
-    "Let us see which images were predicted wrongly:"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 16,