Skip to content

Instantly share code, notes, and snippets.

@manu3193
manu3193 / keras_bottleneck_multiclass.py
Created October 12, 2017 10:37 — forked from Thimira/keras_bottleneck_multiclass.py
Learn how to build a multi-class image classification system using bottleneck features from a pre-trained model in Keras to achieve transfer learning.
'''
Using Bottleneck Features for Multi-Class Classification in Keras
We use this technique to build powerful (high accuracy without overfitting) Image Classification systems with small
amount of training data.
The full tutorial to get this code working can be found at the "Codes of Interest" Blog at the following link,
http://www.codesofinterest.com/2017/08/bottleneck-features-multi-class-classification-keras.html
Please go through the tutorial before attempting to run this code, as it explains how to setup your training data.