时间戳

首页 > 解决方案 > UFF 解析在语句 assert(parser.parse_buffer(stream, 0, network, model_datatype)) 的第 255 行失败

python - UFF 解析在语句 assert(parser.parse_buffer(stream, 0, network, model_datatype)) 的第 255 行失败

问题描述

我想使用库 TensorRT 优化我的模型，但是AssertionError: UFF parsing failed on line 255 in statement assert(parser.parse_buffer(stream, 0, network, model_datatype))当我尝试执行函数时出现错误trt.legacy.utils.uff_to_trt_engine。我是 TensorRT 的新手。告诉我，我做错了什么？谢谢

FROZEN_FPATH = 'frozen.pb'
ENGINE_FPATH = 'engine.plan'
INPUT_NODE = 'the_input_kz'
ONPUT_NODE = 'softmax_kz/truediv'
INPUT_SIZE = [3,1024,1024]
MAX_BATCH_SIZE =1
MAX_WORKSPACE =1<<20

uff_model = uff.from_tensorflow_frozen_model(FROZEN_FPATH, [ONPUT_NODE])

parser =uffparser.create_uff_parser()
parser.register_input(INPUT_NODE,INPUT_SIZE,0)
parser.register_output(ONPUT_NODE)
G_LOGGER = trt.Logger(trt.Logger.INFO)

engine = trt.legacy.utils.uff_to_trt_engine(G_LOGGER, uff_model, parser, MAX_BATCH_SIZE, MAX_WORKSPACE)

执行函数时trt.legacy.utils.uff_to_trt_engine会发出错误

标签： pythontensorflowmachine-learningkerastensorrt

解决方案

推荐阅读