File size: 577 Bytes
391ff22 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | import faulthandler, sys, tensorrt as trt
faulthandler.enable(file=sys.stderr, all_threads=True)
logger = trt.Logger(trt.Logger.VERBOSE)
builder = trt.Builder(logger)
flag = 1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH)
network = builder.create_network(flag)
parser = trt.OnnxParser(network, logger)
with open('eyelike_oob_input.onnx', 'rb') as f:
blob = f.read()
print('about to parse', len(blob), 'bytes', flush=True)
ok = parser.parse(blob)
print('parse() returned', ok)
for i in range(parser.num_errors):
print(' err', i, ':', parser.get_error(i))
|