OpenVINO™ Training Extensions provide a convenient environment to train Deep Learning models and convert them using the OpenVINO™ toolkit for optimized inference.
-
Download and install OpenVINO™.
-
Clone repository in the working directory by running the following:
git clone https://github.com/openvinotoolkit/training_extensions.git --single-branch cd training_extensions
-
Install prerequisites by running the following:
sudo apt-get install python3-pip virtualenv
-
Create and activate virtual environment:
virtualenv venv source venv/bin/activate
-
Install
ote
package:pip3 install -e ote/
After installation, you are ready to train your own models, evaluate and use them for prediction.
Models that were previously developed can be found here.