Skip to content

All useful sample codes of tensorrt models using onnx

License

Notifications You must be signed in to change notification settings

Zulex/TensorRT_Examples

 
 

Repository files navigation

Examples of TensorRT models using ONNX

All useful sample codes of TensorRT models using ONNX

1. Basic step

  1. Generation TensorRT Model by using ONNX
    1.1 TensorRT CPP API
    1.2 TensorRT Python API
    1.3 Polygraphy

  2. Dynamic shapes for TensorRT
    2.1 Dynamic batch
    2.2 Dynamic input size

2. Intermediate step

  1. Custom Plugin
    3.1 Adding a pre-processing layer by cuda

  2. Modifying an ONNX graph by ONNX GraphSurgeon
    4.1 Extracting a feature map of the last Conv for Grad-Cam
    4.2 Generating a TensorRT model with a custom plugin and ONNX

  3. TensorRT Model Optimizer
    5.1 Explict Quantization (PTQ)
    5.2 Explict Quantization (QAT)
    5.3 Sparsity (2:4 sparsity pattern)

3. Advanced step

  1. Super Resolution
    6.1 Real-ESRGAN
  2. Object Detection
  3. Instance Segmentation
  4. Semantic Segmentation
  5. Depth Estimation
    10.1 Depth Pro ( "It is under repair due to an accuracy issue.")

4. reference

About

All useful sample codes of tensorrt models using onnx

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 70.2%
  • C++ 27.3%
  • CMake 1.3%
  • Cuda 1.2%