语句断言（parser.parse_buffer（流、0、网络、模型_数据类型））中第255行的UFF分析失败

Question

我想使用库TensorRT优化我的模型，但在尝试执行excute函数trt.legacy.utils.uff_to_trt_engine时出现错误AssertionError: UFF parsing failed on line 255 in statement assert(parser.parse_buffer(stream, 0, network, model_datatype))。我是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时会发出错误