Book Image

Hands-On Neural Networks with TensorFlow 2.0

By : Paolo Galeone
Book Image

Hands-On Neural Networks with TensorFlow 2.0

By: Paolo Galeone

Overview of this book

TensorFlow, the most popular and widely used machine learning framework, has made it possible for almost anyone to develop machine learning solutions with ease. With TensorFlow (TF) 2.0, you'll explore a revamped framework structure, offering a wide variety of new features aimed at improving productivity and ease of use for developers. This book covers machine learning with a focus on developing neural network-based solutions. You'll start by getting familiar with the concepts and techniques required to build solutions to deep learning problems. As you advance, you’ll learn how to create classifiers, build object detection and semantic segmentation networks, train generative models, and speed up the development process using TF 2.0 tools such as TensorFlow Datasets and TensorFlow Hub. By the end of this TensorFlow book, you'll be ready to solve any machine learning problem by developing solutions using TF 2.0 and putting them into production.
Table of Contents (15 chapters)
Free Chapter
1
Section 1: Neural Network Fundamentals
4
Section 2: TensorFlow Fundamentals
8
Section 3: The Application of Neural Networks

Create a TensorFlow DatasetBuilder

In the same way as any other machine learning problem, the first step is getting the data. Since semantic segmentation is a supervised learning task, we need a classification dataset of images and corresponding labels. The peculiarity is that the label in itself is an image.

At the time of writing, there is no semantic dataset ready to use in TensorFlow Datasets. For this reason, we use this section not only to create tf.data.Dataset with the data that we need, but also to have a look at the process required to develop a tfds DatasetBuilder.

Since, in the previous section dedicated to the object detection, we used the PASCAL VOC 2007 dataset, we are going to reuse the downloaded files to create the semantic segmentation version of the PASCAL VOC 2007 dataset. The following screenshot shows how the dataset is provided. Each picture has a corresponding...