OpenVINO™ Training Extensions

OpenVINO™ Training Extensions provide a suite of advanced algorithms to train Deep Learning models and convert them using the OpenVINO™ toolkit for optimized inference. It allows you to export and convert the models to the needed format. OpenVINO Training Extensions independently create and train the model. It is open-sourced and available on GitHub. Read the OpenVINO Training Extensions documentation to learn more.

Detailed Workflow

_images/training_extensions_framework.png
  1. To start working with OpenVINO Training Extensions, prepare and annotate your dataset. For example, on CVAT.

  2. OpenVINO Training Extensions train the model, using training interface, and evaluate the model quality on your dataset, using evaluation and inference interfaces.

    Note

    Prepare a separate dataset or split the dataset you have for more accurate quality evaluation.

  3. Having successful evaluation results received, you have an opportunity to deploy your model or continue optimizing it, using NNCF and POT. For more information about these frameworks, go to Optimization Guide.

If the results are unsatisfactory, add datasets and perform the same steps, starting with dataset annotation.