googlenet-v4-tf

Use Case and High-Level Description

The googlenet-v4-tf model is the most recent of the Inception family of models designed to perform image classification. Like the other Inception models, the googlenet-v4-tf model has been pretrained on the ImageNet image database. Originally redistributed as a checkpoint file, was converted to frozen graph. For details about this family of models, check out the paper, repository.

Steps to Reproduce Conversion to Frozen Graph

  1. Clone the original repository
    git clone https://github.com/tensorflow/models.git
    cd models/research/slim
  2. Checkout the commit that the conversion was tested on:
    git checkout 5d36f19
  3. Apply freeze.py.patch patch
    git apply path/to/freeze.py.patch
  4. Download the pretrained weights
  5. Install the dependencies:
    pip install tensorflow==1.14.0
  6. Run
    python3 freeze.py --ckpt path/to/inception_v4.ckpt --name inception_v4 --num_classes 1001 --output InceptionV4/Logits/Predictions

Example

Specification

Metric Value
Type Classification
GFLOPs 24.584
MParams 42.648
Source framework TensorFlow*

Accuracy

Metric Original model Converted model
Top 1 80.21% 80.21%
Top 5 95.20% 95.20%

Performance

Input

Original model

Image, name - input, shape - 1,299,299,3, format is B,H,W,C where:

  • B - batch size
  • H - height
  • W - width
  • C - channel

Channel order is RGB. Mean values - [127.5, 127.5, 127.5], scale value - 127.5

Converted model

Image, name - data, shape - 1,3,299,299, format is B,C,H,W where:

  • B - batch size
  • C - channel
  • H - height
  • W - width

Channel order is BGR

Output

Original model

Object classifier according to ImageNet classes, name - InceptionV4/Logits/Predictions, shape - 1,1001, output data format is B,C where:

  • B - batch size
  • C - Predicted probabilities for each class in [0, 1] range

Converted model

Object classifier according to ImageNet classes, name - InceptionV4/Logits/Predictions, shape - 1,1001, output data format is B,C where:

  • B - batch size
  • C - Predicted probabilities for each class in [0, 1] range

Legal Information

The original model is distributed under the Apache License, Version 2.0. A copy of the license is provided in APACHE-2.0-TensorFlow.txt.