Class wise Threshold

Hi,

I just want to know how I can use class wise confidence threshold.

Can you describe the use case and the question in details? thank you.

actually I have 3 classes annotations in which my darknet model trained on. now while using this model for inferencing in deepstream on a video I want to use class wise threshold I don’t want to use single confidence threshold value for all the 3 classes. Is there any way through which I can set class wise confidence threshold in deepstream config or in nvinfer config.

You can try to set the [class-attrs-<class-id>] group to control your classes.
https://docs.nvidia.com/metropolis/deepstream/dev-guide/text/DS_plugin_gst-nvinfer.html

1 Like

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.