Deep Learning Tutorial: Build an Image Recognizer on your dataset in less than 10 minutesSalim ChemlalBlockedUnblockFollowFollowingApr 26Photo by Andre Mouton on UnsplashThis tutorial is a concise step-by-step guide for building and training an Image Recognizer on any image dataset of your choice.
In this tutorial, you will learn how to:Scrape images from Google Images and create your own datasetBuild and train an image recognizer on your datasetVisualize and adequately interpret classification resultsTest model with new imagesTo run this notebook, you can simply open it with Google Colab here.
Once in Colab, make sure to change the following to enable GPU backend,Runtime -> Change runtime type -> Hardware Accelerator -> GPU1.
Image Dataset Download and SetupOption 1: Working with a ready dataset of your ownIf you would like to use your own image dataset for this tutorial, rearrange it in a way that images of the same class are under the same folder.
Then, name the folders with the corresponding class labels.
Option 2: Scraping images from Google ImagesIf you do not have a dataset in-hand, you can scrape images from Google Images and make up a dataset of your choice.
In order to do so, simply install Fatkun Batch Download Image extension on your google chrome browser and download any set of google images by clicking the extension tab.
Place all images of the same class under the same folder and name it accordingly with the class label.
Option 3: Working with my Monkeys dataset, which I originally scraped from Google Images :)The dataset can be downloaded and untarred using the following couple bash commands.
In this tutorial, we will be classifying Monkeys, so we created 6 folders corresponding to 6 different types of monkeys, as shown below.
If you choose to run this notebook with your dataset, simply replace the url and file name in the above cell with yours.
Storing the .
tar file in Dropbox worked best for me.
2.
Image RecognitionFor a further explanation of the code and functions used within this notebook, check out my other post: Concise Notes on Image Recognition, Fastai 2019 v3 Lesson 1InitializationPATH is the path containing all the class folders.
You may keep ‘.
/Monkeys Faces’ if you are using Monkeys dataset.
Otherwise, specify the path of your folders.
Let’s retrieve the image classes,There are 6 classes: ['Macaque', 'Orungatun', 'Chimpanzee', 'Gorilla', 'Mandrill', 'Gibbon']Let’s verify there are not any corrupt images that cannot be read.
If found any, they will simply be deleted.
Creating and training the classifierLet’s create training and validation sets,There are 268 training images and 67 validation images.
Let’s visualize some images from different classes,Let’s build our Deep Convolutional Neural Network (CNN),Let’s pick the appropriate learning rate parameter,Use the above plot to pick adequate learning rates for your model.
We need two learning rates since we are using cyclic learning rates:The first learning rate is just before the loss starts to increase, preferably 10x smaller than the rate at which the loss starts to increase.
For instance, 1e-02 for our Monkeys Face Recognition dataset.
The second learning rate is 10x smaller than the first learning rate, so 1e-03 in our example.
The plot will be different for your dataset, so make sure to pick these two learning rates accordingly.
Replace your chosen learning rates in the slice function below and let’s train the model.
Great, we achieved a high classification accuracy with only a few lines of code and without much tuning of the parameters.
We are DONE but let’s further interpret the results.
3.
Results Interpretation and VisualizationWe can start by visualizing a confusion matrix.
The diagonal elements represent the number of images for which the predicted label is equal to the true label, while off-diagonal elements are those that are mislabeled by the classifier.
We can also plot images with top losses; in other words, the images that the model was most confused about.
A high loss implies high confidence about the wrong answer.
Images are shown along with top losses:prediction label / actual label / loss / probability of actual image class.
4.
Testing the model on a new imageLet’s now feed the model a new image it never seen before and see how well it classifies it.
Upload the image to the same folder of this notebook.
It is a MandrillCongratulations completing, reading, or skimming through this tutorial.
Acknowledgment: This tutorial uses fastai library, so we thank Jeremy Howard and fast.
ai.
.