README.md

Training Toolbox for TensorFlow*

Training Toolbox for TensorFlow* provides a convenient environment to train Deep Learning models and convert them using OpenVINO™ Toolkit for optimized inference.

Prerequisites

• Ubuntu* 16.04 / 18.04
• Python* 3.4-3.6
• libturbojpeg
• For Python prerequisites, refer to requirements.txt
• (Optional) TensorFlow GPU prerequisites.
• (Optional) OpenVINO™ R3 to export trained models

Setup

Install requirements:

pip3 install cython
sudo apt install 2to3 protobuf-compiler

Models

After installation, you are ready to train your own models, evaluate them, and use them for predictions.

• Action Detection
• Image Retrieval
• License Plate Recognition
• Person Vehicle Bike Detector
• SSD MobileNet FPN 602
• SSD Object Detection
• Text Detection
• Text Recognition
• Vehicle Attributes
• Image Retrieval