TensorRT 4.0 produces the following error when I try to create TensorRT engine from a UFF model:
[TensorRT] ERROR: conv1_1/Conv2D: kernel weights has count 1152 but 282816 was expected [TensorRT] ERROR: UFFParser: Parser error: conv1_1/BiasAdd: The input to the Scale Layer is required to have a minimum of 3 dimensions.
I have 2 image inputs of size 435x491, my engine creation:
parser.register_input("Placeholder", (1,435,491), 0) parser.register_input("Placeholder_1", (1,435,491), 0) parser.register_output("outconv/Relu") engine = trt.utils.uff_to_trt_engine(G_LOGGER, uff_model, parser, 1, 1 << 20)
What am I doing wrong?