This article was originally published by Himanshu Sharma on towords data science.
Image classification basically helps us in classifying images into different labels. It is like bucketing different images into the bucket they belong to. For, e.g. a model trained to identify the image of a cat and a dog will help in segregating different images of cats and dogs respectively. There are multiple deep learning frameworks like Tensorflow, Keras, Theano, etc that can be used to create image classification models. Today we will create an image classification model from scratch using Keras and Tensorflow.
Creating image-related modeling can be done using CNN. Convolutional Neural Networks are mainly used for image-related modeling. It is one of the easiest ways to perform image classification, image detection, image segmentation, etc. It contains different types of convolutional layers that help in filtering out the most important features of the image using kernels of different sizes. Some of the most important layers are:
It is used to create a Convolutional kernel that is convolved with the input layer to produce the output tensor.
It is a downsampling technique that takes the maximum value by pool size.
It flattens the input and creates an1-D output.
There are multiple hyper-parameters that can be used accordingly to improve the model performance. These hyper-parameters may include a number of neurons, kernel size, pool size, activation function, etc.
In this article, we will create a network using CNN from scratch. I will show you how you can load data from online sources, preprocess that data and make it ready for modeling, and finally design the model architecture.
Let’s get started…
We will start by installing all required libraries. We will install Keras, TensorFlow and will also install TensorBay (SDK on Graviti data platform） that we will use for loading the dataset. The command given below will do that.
In this step, we will import all the required libraries and functions to load the data, preprocess it and create the model.
This is the initial step, where we will load the data from TensorBay. In order to download a dataset from TensorBay, we need to create an account and fork a dataset on which we want to work, it contains a large variety of image datasets. For loading the dataset in our Jupyter notebook we need to have the access key to TensorBay that we can download from developer tools in our account.
For this article, I am using the 17 Category Flower, as you can see I have loaded it above using TensorBay.
gas = GAS("<Your Key>") dataset = Dataset("Flower17", gas) segment = dataset
In this step, we will preprocess the data and make it ready for modeling. We will start by creating the image and labels list and loading data into it.
Next, let us visualize the number of classes that we have in this dataset.
Next, we will split the dataset, normalize it and also binarize the labels.
This is the final step, where we will create the model architecture, compile the model and train it. Before creating the model architecture, let’s split the training data into train and validation.
As you can see above we have created and trained the model using some hyper-parameters, model accuracy is not that good but you can always tune the hyper-parameters to increase the performance. We can also save the model by using the command given below.
Go ahead, try this with different datasets and create CNN models easily by following this article. If you want to learn more about Graviti data platform, check out another tutorial: A quick guide to OCR with Transformer.