summaryrefslogtreecommitdiff
path: root/Content/posts/2019-12-10-TensorFlow-Model-Prediction.md
diff options
context:
space:
mode:
Diffstat (limited to 'Content/posts/2019-12-10-TensorFlow-Model-Prediction.md')
-rw-r--r--Content/posts/2019-12-10-TensorFlow-Model-Prediction.md60
1 files changed, 60 insertions, 0 deletions
diff --git a/Content/posts/2019-12-10-TensorFlow-Model-Prediction.md b/Content/posts/2019-12-10-TensorFlow-Model-Prediction.md
new file mode 100644
index 0000000..cafa026
--- /dev/null
+++ b/Content/posts/2019-12-10-TensorFlow-Model-Prediction.md
@@ -0,0 +1,60 @@
+---
+date: 2019-12-10 11:10
+description: Making predictions for image classification models built using TensorFlow
+tags: Tutorial, Tensorflow, Code-Snippet
+---
+
+# Making Predictions using Image Classifier (TensorFlow)
+
+*This was tested on TF 2.x and works as of 2019-12-10*
+
+If you want to understand how to make your own custom image classifier, please refer to my previous post.
+
+If you followed my last post, then you created a model which took an image of dimensions 50x50 as an input.
+
+First we import the following if we have not imported these before
+
+```python
+import cv2
+import os
+```
+
+Then we read the file using OpenCV.
+
+```python
+image=cv2.imread(imagePath)
+```
+
+The cv2. imread() function returns a NumPy array representing the image. Therefore, we need to convert it before we can use it.
+
+```python
+image_from_array = Image.fromarray(image, 'RGB')
+```
+
+Then we resize the image
+
+```python
+size_image = image_from_array.resize((50,50))
+```
+
+After this we create a batch consisting of only one image
+
+```python
+p = np.expand_dims(size_image, 0)
+```
+
+We then convert this uint8 datatype to a float32 datatype
+
+```python
+img = tf.cast(p, tf.float32)
+```
+
+Finally we make the prediction
+
+```python
+print(['Infected','Uninfected'][np.argmax(model.predict(img))])
+```
+
+`Infected`
+
+