Print confidence score with bbox during luxonis_train infer

shail

Hi, maybe a simpe question but is there a way to also print/display the confidence score along with the bounding box and class label when running the luxonis_train infer command on test view or on custom dataset directory? Currently I only see the bbox and class name on the output visualizations. If not, what is the recommended way to access per-prediction confidence scores programmatically?

Thanks!

KlemenSkrlj

Hi @shail
This is currently not supported yet but it is a very good feature request. We have made a PR for it here so you can test it out. The way to add the scores visualization as well is to add draw_scores: True to the visualizer config params.
Let me know if that works.
Klemen

shail

@KlemenSkrlj Thank you for getting back. I will test it out next time when I am running an inference and will let you know how it goes.

KlemenSkrlj

Sounds good. Note that this also works for training or testing, if you set these parameters through config the inference images will also have the scores displayed.

shail

Thanks klemen for this feature!