Skip to content

Instantly share code, notes, and snippets.

View olaysco's full-sized avatar
🏠
Working from home

Olayiwola Odunsi olaysco

🏠
Working from home
View GitHub Profile
@olaysco
olaysco / keras_bottleneck_multiclass.py
Created April 9, 2018 14:24 — 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.