Show all classes probabilities

If I want for each bounding box to get all the probabilities for all the classes and not only for the “winner” class. Is there a way to do it somehow?

May I know which network did you mention?

Sure, I use Yolo v4 via nvidia tao

It is possible to get the confidence when run inference with the tensorrt engine.
For example, if you run inference via trition-app, you can take a look at the yolov3 inference via triton-app. From tao-toolkit-triton-apps/ at main · NVIDIA-AI-IOT/tao-toolkit-triton-apps · GitHub , the confidence can be found for all the classes.
If you run inference with deepstream, you can refer to deepstream_tao_apps/nvdsinfer_custombboxparser_tao.cpp at master · NVIDIA-AI-IOT/deepstream_tao_apps · GitHub

1 Like

How can i use Tao with yolor detection

Please create a new forum topic with more detailed info. Thanks.

1 Like

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