Training_Extensions
OpenVINO™ Training Extensions is an open-source toolkit developed by Intel designed for training, evaluating, and deploying deep learning models optimized for OpenVINO inference. It supports a range of computer vision tasks including classification, object detection, semantic and instance segmentation, and anomaly recognition. The toolkit provides validated model templates, or "recipes," which consolidate necessary configurations and have been tested on various datasets to offer reliable starting points for model development. Users prepare datasets, train models via a command-line interface, evaluate on validation sets, and export models in OpenVINO IR or ONNX formats for deployment. The toolkit supports native Intel GPU (XPU) training and testing, distributed training across multiple GPUs, mixed-precision training, and class incremental learning to add new classes to existing models. It integrates with tools like NNCF for post-training optimization and supports multiple backends starting from version 2.4.5. OpenVINO Training Extensions is primarily focused on computer vision tasks and is available as a free, open-source solution.
OpenVINO Training Extensions is an open-source Intel toolkit for training and deploying computer vision models optimized for OpenVINO inference.
Custom Computer Vision Model Training
Developers can train custom models for tasks like object detection or segmentation optimized for Intel hardware.
Model Deployment on Intel Hardware
Export models in OpenVINO IR or ONNX formats for optimized inference on Intel GPUs and other supported devices.
Incremental Class Addition
Add new classes to existing models using class incremental learning without full retraining.
git clone https://github.com/open-edge-platform/training_extensions to obtain the source code.python -m venv venv; source venv/bin/activate; pip install -e ote_cli/ -c external/model-preparation-algorithm/constraints.txt.ote train --train-data-roots --train-ann-file --val-data-roots --val-ann-files --save-model-to to start training.