I’m tring to convert the tensorflow model zoo implementation of maskRcnn to TensorRT using the onnx parser.
running in ngc continer TensorRT19:02 ( tensorRT v 5.0.2 ) I’m using cuda 10.0 on gtx 1050.
I’ve converted the the save model to onnx format using the current available https://github.com/onnx/tensorflow-onnx version 1.8, using opset 6. ( which is officially supported by tensorRT v 5.0.2)
when I load the generated onnx file i get the error “Unsupported ONNX data type: UINT8 (2)”
code to load the onnx file is taken from nvidia docs:
import tensorrt as trt
TRT_LOGGER = trt.Logger(trt.Logger.WARNING)
with trt.Builder(TRT_LOGGER) as builder, builder.create_network() as network, trt.OnnxParser(network, TRT_LOGGER) as parser:
with open(my onnx file, ‘rb’) as model:
(i’ve found https://github.com/onnx/onnx-tensorrt/issues/400
which suggests to use onnx-surgeon to edit the onnx file, this lead to the following error during parsing " [TensorRT] ERROR: Parameter check failed at: …/builder/Network.cpp::addInput::406, condition: isValidDims(dims)" on which i have not found any suggestions online )