[urgent] How to save the bbox confidence probability (i.e. prediction score) during faster-rcnn inference?

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Hi there,

If I run inference with tlt-infer faster_rcnn with bbox_caption_on: True, it saves images with probability, but it doesn’t save in the label files. How can I save the bbox confidence probability (i.e. prediction score) in the labels files during faster-rcnn inference?

Thanks

3

Hi cogbot,
Unfortunately, the faster-rcnn inference does not support saving prediction score in the label file. I will sync with infernal team for your request.

4

Please. Thanks, @Morganh.

5

The feature will be available in next release.

6

@Morganh, Thanks, again for the update. That’s great. When will we have the next release?

7

It is expected within this year.

1 Like
8

For other network, like detectnet_v2, ssd, dssd,retinanet, yolo, they already have this feature.

1 Like