TensorRT incorrect predictions for segmentation

I am trying to do a segmentation model tensorrt inference. It is giving incorrect results.
(tensorrt version: ‘’)

Attached the necessary files to reproduce the issue.
cmap_cityscapes.npy (356 Bytes)
model.onnx (11.5 MB)
model_fp32.engine (12.5 MB)
Nvidia_TensorRT.ipynb (48.3 KB)

Hi @dheerajmadda ,
Apologies for the miss,
I am trying to reproduce the issue from my end, Will update you soon.


1 Like

Any update on this please? The tensorRT is producing zeros as outputs.