I have been working on an inference script for a segmentation model on the NVIDIA Driveworks 3.5 Platform. As of now I have converted the segmentation model to the tensorrt format, loaded the model, and been able to run inference using the line:
CHECK_DW_ERROR(dwDNN_inferRaw(m_dnnOutputsDevice, &m_dnnInputDevice, 1U, m_dnn));
I now need to display only the second channel of this output tensor. My queries are:
- Do i need to stream the output tensor to CPU to perform any tensor operations?
- Is there a way to convert a tensor to an image handle that would allow me to stream it to GL directly.
- What would be the best approach to retrieve only a particular channel of the output tensor, and display it on the gl renderer.