Training Toolbox for TensorFlow provides a convenient environment to train Deep Learning models and convert them using OpenVINO™ Toolkit for optimized inference.
- Ubuntu 16.04 / 18.04
- Python 3.4-3.6
- libturbojpeg
- For Python pre-requisites refer to
requirements.txt - (Optional) TensorFlow GPU pre-requisites.
- (Optional) OpenVINO™ R3 for exporting of the trained models
- Install requirements
pip3 install cython
sudo apt install 2to3 protobuf-compiler- Download submodules
cd openvino_training_extensions
git submodule update --init --recommend-shallow external/cocoapi external/models- Compile Protobuf libraries
cd openvino_training_extensions/external/models/research/
protoc object_detection/protos/*.proto --python_out=.After installation, you are ready to train your own models, evaluate them, use them for predictions.