I am trying to run inference on Jetson TX2 for an image segmentation DNN. The network was written in TensorFlow. I was able to convert the trained network to UFF file, but when I try to create TensorRT engine following SampleUffMnist example (replacing the UFF file with mine and changing input and output node names), I get this error:
ERROR: UFFParser: Parser error: Reshape_1: Reshape: Volume mismatch ERROR: sample_uff_mnist: Fail to parse ERROR: sample_uff_mnist: Model load failed
What does this error mean and how can I resolve it?