1 Star 0 Fork 0

jianfeng/TensorRT

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

TensorRT Samples

Contents

1. "Hello World" Samples

Sample Language Format Description
sampleOnnxMNIST C++ ONNX “Hello World” For TensorRT With ONNX
sampleMNIST C++ Caffe “Hello World” For TensorRT Using Caffe
end_to_end_tensorflow_mnist Python UFF “Hello World” For TensorRT Using TensorFlow
network_api_pytorch_mnist Python INetwork “Hello World” For TensorRT Using Pytorch

2. TensorRT API Samples

Sample Language Format Description
sampleAlgorithmSelector C++ Caffe Algorithm Selection API usage
sampleCharRNN C++ INetwork Building An RNN Network Layer By Layer
sampleDynamicReshape C++ ONNX Digit Recognition With Dynamic Shapes In TensorRT
sampleINT8 C++ Caffe Performing Inference In INT8 Using Custom Calibration
sampleINT8API C++ Caffe Performing Inference In INT8 Precision
sampleMNISTAPI C++ INetwork Building a Simple MNIST Network Layer by Layer
sampleOnnxMnistCoordConvAC C++ ONNX Implementing CoordConv with a custom plugin
sampleIOFormats C++ Caffe Specifying TensorRT I/O Formats
sampleUffMNIST C++ UFF Import A TensorFlow Model And Run Inference
sampleUffPluginV2Ext C++ UFF Adding A Custom Layer That Supports INT8 I/O To Your Network
trtexec C++ All TensorRT Command-Line Wrapper: trtexec
engine_refit_mnist Python INetwork Refitting A TensorRT Engine
int8_caffe_mnist Python Caffe INT8 Calibration
introductory_parser_samples Python Any Introduction To Importing Models Using TensorRT Parsers
onnx_packnet Python ONNX TensorRT Inference Of ONNX Models With Custom Layers
uff_custom_plugin Python INetwork Adding A Custom Layer To Your TensorFlow Network In TensorRT

3. Application Samples

Sample Language Format Description
sampleFasterRCNN C++ Caffe Object Detection With Faster R-CNN
sampleUffFasterRCNN C++ UFF Object Detection With A TensorFlow FasterRCNN Network
sampleGoogleNet C++ Caffe Building And Running GoogleNet In TensorRT
sampleSSD C++ Caffe Object Detection With SSD
sampleUffSSD C++ UFF Object Detection With A TensorFlow SSD Network
sampleUffMaskRCNN C++ UFF Object Detection And Instance Segmentation With MasK R-CNN Network
efficientdet Python ONNX EfficientDet Object Detection with TensorRT
efficientnet Python ONNX EfficientNet V1 and V2 Classification with TensorRT
tensorflow_object_detection_api Python ONNX TensorFlow Object Detection API Models in TensorRT
uff_ssd Python UFF Object Detection with SSD
yolov3_onnx Python ONNX Object Detection Using YOLOv3 With TensorRT ONNX Backend

Known Limitations

  • UFF converter and GraphSurgeon tools are only supported with Tensorflow 1.x
  • For the UFF samples, please use the NVIDIA tf1 (Tensorflow 1.x) for running these tests or install Tensorflow 1.x manually.
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/ivofeng/TensorRT.git
git@gitee.com:ivofeng/TensorRT.git
ivofeng
TensorRT
TensorRT
main

搜索帮助