iwatake2222 / play_with_tensorrt

Sample projects for TensorRT in C++
Apache License 2.0
190 stars 34 forks source link
cpp deep-learning jetson nvidia tensorrt

Play with TensorRT

Target

Usage

./main [input]

 - input = blank
    - use the default image file set in source code (main.cpp)
    - e.g. ./main
 - input = *.mp4, *.avi, *.webm
    - use video file
    - e.g. ./main test.mp4
 - input = *.jpg, *.png, *.bmp
    - use image file
    - e.g. ./main test.jpg
 - input = number (e.g. 0, 1, 2, ...)
    - use camera
    - e.g. ./main 0
- input = jetson
    - use camera via gstreamer on Jetson
    - e.g. ./main jetson

How to build a project

0. Requirements

1. Download

2-a. Build in Linux

2-b. Build in Windows (Visual Studio)

Configuration for TensorRT

You don't need to change any configuration for TensorRT, but you can change it if you want.

Model format

DLA Cores (NVDLA)

Model conversion settings

Quantization Calibration

Note

License

Acknowledgements