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 tf import numpy as np import 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_data fashion_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)) print("No of labels in test set {}".format(fashion_mnist.test.labels.shape))
No of images in training set (55000, 784) No of labels in training set (55000, 10) No of images in test set (10000, 784) No of labels in test...