Keras Multi label Image Classification
The objective of this study is to develop a deep learning model that will identify the natural scenes from images. This type of problem comes under multi label image classification where an instance can be classified into multiple classes among the predefined classes. https://suraj-deshmukh.github.io/Keras-Multi-Label-Image-Classification/
Dataset
The complete description of dataset is given on http://lamda.nju.edu.cn/data_MIMLimage.ashx. The dataset contains 2000 natural scenes images.
Keras Model Architecture
Preprocessing
Resized all images to 100 by 100 pixels and created two sets i.e train set and test set. Train set contains 1600 images and test set contains 200 images.
Training
As this is multi label image classification, the loss function was binary crossentropy and activation function used was sigmoid at the output layer. Keras doesn't have provision to provide multi label output so after training there is one probabilistic threshold method which find out the best threshold value for each label seperately, the performance of threshold values are evaluated using Matthews Correlation Coefficient and then uses this thresholds to convert those probabilites into one's and zero's.
Result
Below table shows the result on test set
Accuracy
Value
Hamming loss
0.1395
Exact Match
0.54
Preprocessed Dataset and Weight file download link
Ipython notebook
Visualization
Command:
python visualization.py
link: localhost:5000
Keras repo: