Skip to content

Latest commit

 

History

History
 
 

pytorch_toolkit

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
face_recognition
face_recognition
 
 
human_pose_estimation
human_pose_estimation
 
 
instance_segmentation
instance_segmentation
 
 
README.md
README.md
 
 

README.md

Training Toolbox for PyTorch

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

Pre-requisites

  • Ubuntu 16.04 / 18.04
  • Python 3.4-3.6
  • libturbojpeg
  • For Python pre-requisites refer to requirements.txt
  • (Optional) OpenVINO™ R3 for exporting of the trained models

Quick Start Guide

Setup Training Toolbox for PyTorch

  1. Create virtual environment 
cd /<path_to_working_dir>/training_toolbox/pytorch_toolkit/<model>
bash init_venv.sh
  1. Start to work
. venv/bin/activate

Note: if you have installed the OpenVino toolkit after creating a virtual environment then you have to recreate one to install required packages for the Model Optimizer into one.

Do not forget to update several environment variables are required to compile and run OpenVINO™ toolkit applications, for details see: https://software.intel.com/en-us/articles/OpenVINO-Install-Linux.

Models

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