___ _ _ __ ______ ___ _
|_ _|_ _ __ ___ _ __| |_(_)___ _ \ \ / /__ / | _ |_)_ _ __ _ _ _ _ _
| || ' \/ _/ -_) '_ \ _| / _ \ ' \ V / |_ \ | _ \ | ' \/ _` | '_| || |
|___|_||_\__\___| .__/\__|_\___/_||_\_/ |___/ |___/_|_||_\__,_|_| \_, |
|_| |__/
This project is based on the lesson of Francois Chollet
All the data is available on Kaggle
My tutorial
Keras implementation of InceptionV3 convolutional neural network in solving binary classification problems
The folder should be like this:
data/
train/
dogs/
dog001.jpg
dog002.jpg
...
cats/
cat001.jpg
cat002.jpg
...
validation/
dogs/
dog001.jpg
dog002.jpg
...
cats/
cat001.jpg
cat002.jpg
...
Ordering:
- bottleneck_features.py: Fetching the bottleneck features from pretrained InceptionV3 as numpy arrays;
- top_model.py: Training FFN model with numpy arrays;
- complete_model.py: Training the complete model with images;
- visualization.py: Visualize the result.
This code has been successfully tested on:
- Windows 10, Anaconda2-4.3.0.1, Keras 1.1.2, Theano 0.8.2
- Ubuntu 14.04.5, Anaconda2-4.3.0.1, Keras 1.1.2, tensorflow 1.0.0