Converting TensorFlow Slim Image Classification Model Library Models¶
TensorFlow-Slim Image Classification Model Library is a library to define, train and evaluate classification models in TensorFlow. The library contains Python scripts defining the classification topologies together with checkpoint files for several pre-trained classification topologies. To convert a TensorFlow-Slim library model, complete the following steps:
Download the TensorFlow-Slim models git repository.
Download the pre-trained model checkpoint.
Export the inference graph.
Convert the model using model conversion API.
The Example of an Inception V1 Model Conversion below illustrates the process of converting an Inception V1 Model.
Example of an Inception V1 Model Conversion¶
This example demonstrates how to convert the model on Linux OSes, but it could be easily adopted for the Windows OSes.
Step 1. Create a new directory to clone the TensorFlow-Slim git repository to:
git clone https://github.com/tensorflow/models.git tf_models
Step 2. Download and unpack the Inception V1 model checkpoint file:
tar xzvf inception_v1_2016_08_28.tar.gz
Step 3. Export the inference graph — the protobuf file (
.pb) containing the architecture of the topology. This file does not contain the neural network weights and cannot be used for inference.
python3 tf_models/research/slim/export_inference_graph.py \ --model_name inception_v1 \ --output_file inception_v1_inference_graph.pb
Model conversion API comes with the summarize graph utility, which identifies graph input and output nodes. Run the utility to determine input/output nodes of the Inception V1 model:
python3 <PYTHON_SITE_PACKAGES>/openvino/tools/mo/utils/summarize_graph.py --input_model ./inception_v1_inference_graph.pb
The output looks as follows:
1 input(s) detected: Name: input, type: float32, shape: (-1,224,224,3) 1 output(s) detected: InceptionV1/Logits/Predictions/Reshape_1
The tool finds one input node with name
float32, fixed image size
(224,224,3) and undefined batch size
-1. The output node name is
Step 4. Convert the model with the model conversion API:
mo --input_model ./inception_v1_inference_graph.pb --input_checkpoint ./inception_v1.ckpt -b 1 --mean_value [127.5,127.5,127.5] --scale 127.5
-b command line parameter is required because model conversion API cannot convert a model with undefined input size.
For the information on why
--scale command-line parameters are used, refer to the Mean and Scale Values for TensorFlow-Slim Models.
Mean and Scale Values for TensorFlow-Slim Models¶
The TensorFlow-Slim Models were trained with normalized input data. There are several different normalization algorithms used in the Slim library. OpenVINO classification sample does not perform image pre-processing except resizing to the input layer size. It is necessary to pass mean and scale values to model conversion API so they are embedded into the generated IR in order to get correct classification results.
The file preprocessing_factory.py contains a dictionary variable
preprocessing_fn_map defining mapping between the model type and pre-processing function to be used. The function code should be analyzed to figure out the mean/scale values.
The inception_preprocessing.py file defines the pre-processing function for the Inception models. The
preprocess_for_eval function contains the following code:
... import tensorflow as tf if image.dtype != tf.float32: image = tf.image.convert_image_dtype(image, dtype=tf.float32) ... image = tf.subtract(image, 0.5) image = tf.multiply(image, 2.0) return image
image is converted to data type tf.float32 and the values in the tensor are scaled to the
[0, 1] range using the tf.image.convert_image_dtype function. Then the
0.5 is subtracted from the image values and values multiplied by
2.0. The final image range of values is
OpenVINO classification sample reads an input image as a three-dimensional array of integer values from the range
[0, 255]. In order to scale them to
[-1, 1] range, the mean value
127.5 for each image channel should be specified as well as a scale factor
Similarly, the mean/scale values can be determined for other Slim models.
The exact mean/scale values are defined in the table with list of supported TensorFlow-Slim models at the Converting a TensorFlow Model guide.