Giter Club home page Giter Club logo

training_extensions's Introduction

OpenVINO™ Training Extensions

python black mypy openvino

DISCLAIMERS: Some features described below are under development (refer to feature/otx branch). You can find more detailed estimation from the Roadmap section below.

Overview

OpenVINO™ Training Extensions (OTE) is command-line interface (CLI) framework designed for low-code deep learning model training. OTE lets developers train/inference/optimize models with a diverse combination of model architectures and learning methods using the OpenVINO™ toolkit. For example, users can train a ResNet18-based SSD (Single Shot Detection) model in a semi-supervised manner without worrying about setting a configuration manually. ote build and ote train commands will automatically analyze users' dataset and do necessary tasks for training the model with best configuration. OTE provides the following features:

  • Provide a set of pre-configured models for quick start
    • ote find helps you quickly finds the best pre-configured models for common task types like classification, detection, segmentation, and anomaly analysis.
  • Configure and train a model from torchvision, OpenVINO Model Zoo (OMZ)
    • ote build can help you configure your own model based on torchvision and OpenVINO Model Zoo models. You can replace backbones, necks and heads for your own preference (Currently only backbones are supported).
  • Provide several learning methods including supervised, semi-supervised, imbalanced-learn, class-incremental, self-supervised representation learning
    • ote build helps you automatically identify the best learning methods for your data and model. All you need to do is to set your data in the supported format. If you don't specify a model, the framework will automatically sets the best model for you. For example, if your dataset has long-tailed and partially-annotated bounding box annotations, OTE auto-configurator will choose a semi-supervised imbalanced-learning method and an appropriate model with the best parameters.
  • Integrated efficient hyper-parameter optimization
    • OTE has an integrated, efficient hyper-parameter optimization module. So, you don't need to worry about searching right hyper-parameters. Through dataset proxy and built-in hyper-parameter optimizer, you can get much faster hyper-parameter optimization compared to other off-the-shelf tools. The hyperparameter optimization is dynamically scheduled based on your resource budget.
  • Support widely-used annotation formats
    • OTE uses datumaro, which is designed for dataset building and transformation, as a default interface for dataset management. All supported formats by datumaro are also consumable by OTE without the need of explicit data conversion. If you want to build your own custom dataset format, you can do this via datumaro CLI and API.

Roadmap

v1.0.0 (1Q23)

  • Installation through PyPI
    • Package will be renamed as OTX (OpenVINO Training eXtension)
  • CLI update
    • Update find command to find configurations of tasks/algorithms
    • Introduce build command to customize task or model configurations
    • Automatic algorihm selection for the train command using the given input dataset
  • Adaptation of Datumaro component as a dataset interface
  • Integrate hyper-parameter optimizations
  • Support action recognition task

v1.1.0 (2Q23)

  • SDK/API update

Repository


Quick start guide

In order to get started with OpenVINO™ Training Extensions see the quick-start guide.


License

Deep Learning Deployment Toolkit is licensed under Apache License Version 2.0. By contributing to the project, you agree to the license and copyright terms therein and release your contribution under these terms.


Issues / Discussions

Please use Issues tab for your bug reporting, feature requesting, or any questions.


Contributing

Please read the Contribution guide before starting work on a pull request.


Known limitations

Training, export, and evaluation scripts for TensorFlow- and most PyTorch-based models from the misc branch are, currently, not production-ready. They serve exploratory purposes and are not validated.


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

training_extensions's People

Contributors

albertvanhouten avatar alexanderdokuchaev avatar alexey-sidnev avatar alexey-trushkov avatar alexsu52 avatar ashwinvaidya17 avatar chuneuny-emily avatar djdameln avatar dmitryagapov avatar druzhkov-paul avatar eugene123tw avatar eunwoosh avatar evgeniya-egupova avatar galyazalesskaya avatar goodsong81 avatar harimkang avatar harmjandevries avatar ilya-krylov avatar jihwaneom avatar kprokofi avatar kshpv avatar leonidbeynenson avatar ljcornel avatar nikita-savelyevv avatar saltykox avatar samet-akcay avatar sovrasov avatar sstrehlk avatar supersoob avatar vinnamkim avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.