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.6+
- libturbojpeg
- For Python prerequisites, refer to
requirements.txt - (Optional) TensorFlow GPU prerequisites.
- (Optional) OpenVINO™ to export trained models
Install requirements:
pip3 install cython
sudo apt install 2to3 protobuf-compilerAfter installation, you are ready to train your own models, evaluate them, and use them for predictions.