Quantizing Image Classification Model

This example demonstrates the use of the Post-training Optimization Tool API for the task of quantizing a classification model. The MobilenetV2 model from TensorFlow* is used for this purpose. A custom DataLoader is created to load the ImageNet classification dataset and the implementation of Accuracy at top-1 metric is used for the model evaluation. The code of the example is available on GitHub.

How to prepare the data

To run this example, you need to download the validation part of the ImageNet image database and place it in a separate folder, which will be later referred as <IMAGES_DIR>. Annotations to images should be stored in a separate .txt file (<IMAGENET_ANNOTATION_FILE>) in the format image_name label.

How to Run the example

  1. Launch Model Downloader tool to download mobilenet-v2-1.0-224 model from the Open Model Zoo repository.

    omz_downloader --name mobilenet-v2-1.0-224
  2. Launch Model Converter tool to generate Intermediate Representation (IR) files for the model:

    omz_converter --name mobilenet-v2-1.0-224 --mo <PATH_TO_MODEL_OPTIMIZER>/mo.py
  3. Launch the example script from the example directory:

    python3 ./classification_example.py -m <PATH_TO_IR_XML> -a <IMAGENET_ANNOTATION_FILE> -d <IMAGES_DIR>

    Optional: you can specify .bin file of IR directly using the -w, --weights options.