Let's build an image classifier with TensorFlow and Python!

Welcome back! TensorFlow is one of the best packages for machine learning with Python, so let’s develop another ML project using this programming language! In this specific project, we’ll be developing out an image classifier with TensorFlow! Now, luckily for us the TensorFlow team has already coded out this project and hosted it on Google Colab, so make sure to check out the link below to run this project within your browser:

Google Colaboratory

With that long introduction out of the way, let’s get to coding!

The Code

Starting off, we want to make sure we have TensorFlow and MatPlotLib installed on our machine, so be sure to install those within your environment (unless you're running this in Google Colab). Next up, we want to import these packages as the following:

import tensorflow as tf
import tensorflow_hub as hub
import requests
from PIL import Image
from io import BytesIO