Сбой анализа UFF в строке 255 в утверждении оператора (parser.parse_buffer (stream, 0, network, model_datatype)) - PullRequest
Новая
       4

Сбой анализа UFF в строке 255 в утверждении оператора (parser.parse_buffer (stream, 0, network, model_datatype))

1 голос
/ 04 марта 2020

Я хотел оптимизировать свою модель, используя библиотеку 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 выдается ошибка

...