/keras-inceptionV4

Keras Implementation of Google's Inception-V4 Architecture (includes Keras compatible pre-trained weights)

Primary LanguagePython

Keras Inception-V4

Keras implementation of Google's inception v4 model with ported weights!

As described in: Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning (Christian Szegedy, Sergey Ioffe, Vincent Vanhoucke, Alex Alemi)

Note this Keras implementation tries to follow the tf.slim definition as closely as possible.

Pre-Trained weights for this Keras model can be found here (ported from the tf.slim ckpt): https://github.com/kentsommer/keras-inceptionV4/releases

You can evaluate a sample image by performing the following (weights are downloaded automatically):

  • $ python evaluate_image.py
Loaded Model Weights!
Class is: African elephant, Loxodonta africana
Certainty is: 0.868498

News

3/30/2017:

  1. I have released a Keras-2 compatible version of this network on the keras-2 branch. Note that this only supports the TensorFlow backend.

  2. Included in the keras-2 branch is the ability to specify whether to include the top layers or not.

  3. Once some issues with automatic weight conversion between backends is fixed in Keras 2, the keras-2 branch will be merged into master and should, at that point, support both Theano and TensorFlow.

Performance Metrics (@Top5, @Top1)

Error rate on non-blacklisted subset of ILSVRC2012 Validation Dataset (Single Crop):

  • Top@1 Error: 19.54%
  • Top@5 Error: 4.88%

These error rates are actually slightly lower than the listed error rates in the paper:

  • Top@1 Error: 20.0%
  • Top@5 Error: 5.0%