raymondlo84 / training_extensions

Trainable models and NN optimization tools

OpenVINO™ Training Extensions

OpenVINO™ Training Extensions provide a convenient environment to train Deep Learning models and convert them using the OpenVINO™ toolkit for optimized inference.

Quick Start Guide

Setup OpenVINO™ Training Extensions

1. Download and install OpenVINO™.

2. Clone repository in the working directory by running the following:

   git clone https://github.com/openvinotoolkit/training_extensions.git --single-branch
   cd training_extensions
   

3. Install prerequisites by running the following:

   sudo apt-get install python3-pip virtualenv
   

4. Create and activate virtual environment:

   virtualenv venv
   source venv/bin/activate
   

5. Install ote package:

   pip3 install -e ote/
   

Models

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

• Action Recognition
  • Gesture Recognition
• Instance Segmentation
  • COCO instance segmentation
• Image classification
  • Custom image classification
• Object Detection
  • Face Detection
  • Horizontal Text Detection
  • Person Detection
  • Person Vehicle Bike Detection
  • Vehicle Detection
• Text Spotting
  • Alphanumeric Text Spotting

Misc

Models that were previously developed can be found here.