Classifying fashion products using CNN

We will now see how to use CNN for classifying fashion products.

First, we will import our required libraries as usual:

import tensorflow as tfimport numpy as npimport matplotlib.pyplot as plt%matplotlib inline

Now, we will read the data. The dataset is available in tensorflow.examples, so we can directly extract the data as follows:

from tensorflow.examples.tutorials.mnist import input_datafashion_mnist = input_data.read_data_sets('data/fashion/', one_hot=True)

We will check what we have in our data:

print("No of images in training set {}".format(fashion_mnist.train.images.shape))print("No of labels in training set {}".format(fashion_mnist.train.labels.shape))print("No of images in test set {}".format(fashion_mnist.test.images.shape)) ...

Get Hands-On Reinforcement Learning with Python 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.