Merge pull request #47 from Resseguie/patch-2

Patch 2
MicrosoftDocs · Aug 12, 2021 · c9af224 · c9af224
2 parents 108cac2 + f5e6ccb
commit c9af224
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/05c - Transfer Learning (Tensorflow).ipynb b/05c - Transfer Learning (Tensorflow).ipynb
@@ -306,7 +306,7 @@
     "def predict_image(classifier, image):\n",
     "    from tensorflow import convert_to_tensor\n",
     "    # The model expects a batch of images as input, so we'll create an array of 1 image\n",
-    "    imgfeatures = img.reshape(1, img.shape[0], img.shape[1], img.shape[2])\n",
+    "    imgfeatures = image.reshape(1, image.shape[0], image.shape[1], image.shape[2])\n",
     "\n",
     "    # We need to format the input to match the training data\n",
     "    # The generator loaded the values as floating point numbers\n",