TensorRT 5/6/7 加载onnx模型
TensorRT5/6加载并转换onnx:def onnx_2_trt(onnx_model_name,trt_model_name): with trt.Builder(G_LOGGER) as builder, builder.create_network() as network, trt.OnnxParser(network, G_LOGGER) as parser: builder.max_batch_size = 1024 builder.
原创
2020-07-27 18:11:49 ·
5703 阅读 ·
8 评论