© Brett Koonce 2021
B. KoonceConvolutional Neural Networks with Swift for Tensorflowhttps://doi.org/10.1007/978-1-4842-6168-2_2

2. MNIST: 2D Neural Network

Brett Koonce1  
(1)
Jefferson, MO, USA
 

In this chapter, we will modify our one-dimensional neural network by adding convolutions to produce our first actual convolutional (2D) neural network and use it to categorize black and white (e.g., MNIST) images again.

Convolutions

Convolutions are a deep area of computer vision theory. At a high level we might think of taking an input image and producing another output image:
[cat] --> [magic black box] --> [dog]
Broadly, for any input image there’s a way to convert it to the target image. At the simplest level we might destroy the source image (e.g., multiply ...

Get Convolutional Neural Networks with Swift for Tensorflow: Image Recognition and Dataset Categorization now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.