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 ·
5740 阅读 ·
8 评论