目录
3.1将TensorRT导入Python
程序
导入TensorRT:
import tensorrt as trt
实施一个日志记录界面,TensorRT通过该界面报告错误,警告和参考消息。以下代码显示了如何实现日志记录接口。在这种情况下,我们取消了提示性消息,仅报告警告和错误。TensorRT Python绑定中包含一个简单的记录器。
TRT_LOGGER = trt.Logger(trt.Logger.WARNING)
3.2 在Python中创建网络定义
使用TensorRT进行推理的第一步是根据您的模型创建一个TensorRT网络。
实现这一目标的最简单方法是使用导入模型TensorRT 解析库(见在Python中使用解析器导入模型,使用Python从Caffe中导入,使用Python从TensorFlow导入