Stepping up: Image Classification using TensorFlow 2.x and TensorFlow Dataset

Photo by Bruno Nascimento on Unsplash

In this tutorial, we will build a network to predict the class of an image. Therefore we use the tf_flowers dataset, which consists of 3,670 and 5 different classes. This tutorial builds upon a tutorial by TensorFlow.

This tutorial follows our well-known workflow:

  1. Loading the data and preparing our datasets

--

--

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store
Jan Bollenbacher

Jan Bollenbacher

I am a PhD Student in the field of Machine Learning, living in Germany