Can't get value from custom binary classifier with single sigmoid node output

Please provide complete information as applicable to your setup.

• Hardware Platform (Jetson / GPU) x86/MXM RTX A2000
• DeepStream Version 6.3
• JetPack Version (valid for Jetson only) -
• TensorRT Version v8503
• NVIDIA GPU Driver Version (valid for GPU only) 530.30.02
• Issue Type( questions, new requirements, bugs) Question
• How to reproduce the issue ? (This is for bugs. Including which sample app is using, the configuration files content, the command line used and other details for reproducing)
• Requirement details( This is for new requirement. Including the module name-for which plugin or for which sample application, the function description)

Hi All,

I am fairly new to Deepstream, so please excuse my beginner question.

I’d like to integrate a custom TRT engine model with a single sigmoid output, which corresponds to a binary classifier (0: class 1, 1: class 2). The model is EfficientNetB1 and I converted it successfully to the engine file on the A2000. I can deserialize the model with ‘nvinfer’ and it runs without problems with frame inputs from ‘streammux’. However, I can’t get hold of the output values - using a source pad callback from the next element from nvinfer I always get (NvDsObjectMeta):

  • class_id: -1
  • confidence: 0.0

NvDsClassifierMeta is also not working. I always get num_label: 0.

This is my nvinfer config:


Note that I had to change the input order because my model expects NHWC.

Am I assuming correctly that I can’t use the standard parsing, but need to implement it myself in order to interprete the single sigmoid output correctly?

Since this seems such a simple and common case, is there maybe an example out there for sigmoid single outputs. I couldn’t find any in the Python and C++ samples.

Thanks lot!

I just came across setting output-tensor-meta=1 and then parsing the output from Gst buffer metadata that should be attached by ‘nvinfer’. I need to test this.

I’m closing this topic due to there is no update from you for a period, assuming this issue was resolved. If still need the support, please open a new topic. Thanks