Training Toolbox for PyTorch* provides a convenient environment to train Deep Learning models and convert them using OpenVINO™ Toolkit for optimized inference.
- Ubuntu* 16.04 / 18.04
- Python* 3.4-3.6
- libturbojpeg
- For Python prerequisites, refer to
requirements.txt - (Optional) OpenVINO™ R3 to export trained models
After installation, you are ready to train your own models, evaluate and use them for prediction.
- Action Recognition
- ASL Recognition
- Face Recognition
- Human Pose Estimation
- Instance Segmentation
- Object Detection
- Eye State Classification
- Segmentation of Thoracic Organs
- Super Resolution
- Text Spotting
Tools are intended to perform manipulations with trained models, like compressing models using Quantization-aware training or sparsity.