Skip to content

Xela0421/edgeai-tensorlab

 
 

Repository files navigation

edgeai-tensorlab

Edge AI model training, quantization, compilation/benchmark & Model Zoo

Notice

Our documentation landing pages are the following:

  • https://www.ti.com/edgeai : Technology page summarizing TI’s edge AI software/hardware products
  • https://github.com/TexasInstruments/edgeai : Landing page for developers to understand overall software and tools offering. Read this before navigating into this repository.
  • Our repositories have been restructured : Several repositories are now packaged as components inside this repository.

Release Notes

Please see the release notes


Components

  • The subcomponents have detailed documentation. In the browser, navigate into the sub-folders to see detailed documentation. Here is a high level overview.
Category Tool/Link Purpose IS NOT
Model training & associated tools edgeai-modelzoo provides collection of pretrained models
ditto Model optimization tools Model optimization tools
- Model surgery: Modifies models with minimal loss in accuracy and makes it suitable for TI device (replaces unsupported operators)
- Model Pruning/sparsity: Induces sparsity during training – only applicable for specific devices
- QAT: Quantization Aware Training to improve accuracy with fixed point quantization
- Does not support Tensorflow
ditto edgeai-benchmark Bring your own model and compile, benchmark and generate artifacts for deployment on SDK with camera, inference and display (using edgeai-gst-apps)
- Comprehends inference pipeline including dataset loading, pre-processing and post-processing
- Benchmarking of accuracy and latency with large data sets
- Post training quantization
- Docker for easy development environment setup
ditto edgeai-torchvision
edgeai-mmdetection
edgeai-yolox
edgeai-mmdetection3d
Training repositories for various tasks
- Provides extensions of popular training repositories (like mmdetection, yolox) with lite version of models
- Does not support Tensorflow
ditto edgeai-datasets Example datasets
ditto Model Maker Command line Integrated environment for training & compilation
- Bring your own data, select a model, perform training and generate artifacts for deployment on SDK
- Backend tool for model composer (early availability of features compared to Model Composer )
- Does not support Bring Your Own Model workflow

About

Edge AI Model Development Tools

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 64.4%
  • Python 33.2%
  • C++ 1.0%
  • Shell 0.8%
  • Cuda 0.2%
  • HTML 0.2%
  • Other 0.2%