From 7273f96694736f5e623e905d788bae8ec62a36a1 Mon Sep 17 00:00:00 2001 From: David Resseguie Date: Wed, 11 Aug 2021 21:15:19 -0400 Subject: [PATCH] update predict_image argument mismatch Found same issue from #46 in transfer learning example --- 05c - Transfer Learning (Tensorflow).ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/05c - Transfer Learning (Tensorflow).ipynb b/05c - Transfer Learning (Tensorflow).ipynb index 0f0358a..ad44dee 100644 --- 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], img.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",