Image Tagger - Multi-Label CNN Image Classification

Smart Hashtag Generation

In today’s article, we are going to perform Multi-Label Image Classification using Convolutional Neural Networks. Machine Learning model trained with such an approach, would be able to generate multiple descriptive labels for a given input image. Such an application could be used to generate hashtags for social media posts, like in the below example (iOS App).

‎Hashtagram

Multi-Class CNN Image Classification

Before we dive into the multi-label classification, let’s start with the multi-class CNN Image Classification, as the underlying concepts are basically the same with only a few subtle differences. If you are completely new to this field, I recommend you start with the following article to learn the basics of this topic.

Image Classifier - Cats🐱 vs Dogs🐶

Multi-Class vs Multi-Label Classification

To learn what is a multi-label classification, let’s use an analogy to the multi-class classification and illustrate it with the following example.

Multi-Class

Multi-Label

What’s the difference?

The difference between classes and labels is that…