标签:network python create parser --- logger 序列化 trt Logger
import tensorrt as trt
#build logger
trt_logger=trt.Logger(trt.Logger.Warning)
#create network
#1.parser
dataType=trt.float32
With trt.Builder(trt_logger) as builder,builder.create_network() as
network,trt.CaffeParser() as parser:
Model_tensors=parser.parse(
deploy="sample.prototxt",
model="sample_weight.caffemodel",
network=network,dtype=datatype)
#using onnx to create network
EXPLICIT_BATCH = 1 << (int)(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH)
with trt.Builder(TRT_LOGGER) as builder, builder.create_network(EXPLICIT_BATCH) as
network, trt.OnnxParser(network, TRT_LOGGER) as parser:
with open(model_path, 'rb') as model:
if not parser.parse(model.read()):
for error in range(parser.num_errors):
print(parser.get_error(error))
#创建engine:
Config=builder.create_builder_config()
Config.max_workspace_size=1<<20
With builder.build_engine(network,config) as engine:
#engine 序列化:
Str=engine.serialize()
With open("sample.engine","wb") as f:
f.write(engine.serialize())
标签:network,python,create,parser,---,logger,序列化,trt,Logger 来源: https://blog.csdn.net/qq_27582707/article/details/114503647
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。