使用python API的优点是方便于数据预处理以及后续处理,可以使用numpy和scipy。但是python API的安全性不如C++ API。

使用Python API,结构较为清晰,分为四部:

1、Creating A Network Definition In Python (用python定义网络)

2、Building An Engine In Python

3、Serializing A Model In Python

4、Performing Inference In Python

 

Import TensorRT

import tensorrt as trt

logging interface :  reports errors, warnings, and informational messages

TRT_LOGGER = trt.Logger(trt.Logger.WARNING)

Creating A Network Definition

1、From Scratch

2、Importing From ONNX

相关文章:

  • 2021-06-04
  • 2021-07-27
  • 2022-12-23
  • 2021-12-10
  • 2022-12-23
  • 2022-01-10
  • 2022-01-12
  • 2021-07-29
猜你喜欢
  • 2021-05-05
  • 2021-11-04
  • 2021-07-09
  • 2022-02-02
  • 2021-07-04
  • 2022-01-19
相关资源
相似解决方案