liuyicai/openvino_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. Clone repository in the working directory by running the following:

   git clone https://github.com/opencv/openvino_training_extensions.git
   cd openvino_training_extensions
   

2. Install prerequisites by running the following:

   sudo apt-get install libturbojpeg python3-tk python3-pip virtualenv
   

Models

• PyTorch*

  • Action Recognition
  • ASL Recognition
  • Object Re-Identification
    • Face Recognition
    • Person Re-Identification
    • Vehicle Re-Identification
  • Human Pose Estimation
  • Instance Segmentation
  • Object Detection
    • Face Detection
    • Horizontal Text Detection
    • Person Detection
    • Person Vehicle Bike Detection
    • Vehicle Detection
  • Eye State Classification
  • Segmentation of Thoracic Organs
  • Super Resolution
  • Text Spotting

• TensorFlow*

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

Tools

• PyTorch*

  • Neural Networks Compression Framework - versions <= 1.3 only, for more recent versions check the new NNCF repository

---

* Other names and brands may be claimed as the property of others.