krystiankoziol/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 OpenVINO™ Toolkit for optimized inference.

Quick Start Guide

Setup OpenVINO Training Extensions

1. Clone repository in the working directory

cd /<path_to_working_dir>
git clone https://github.com/opencv/openvino_training_extensions.git

2. Install prerequisites

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

Models

• PyTorch

  • Action recognition
  • Face recognition
  • Human pose estimation
  • Instance segmentation
  • Object Detection
    • Face Detection
    • Person Vehicle Bike Detector
  • Segmentation of thoracic organs
  • Super resolution

• 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

Tools

• PyTorch

  • Neural Networks Compression Framework