NMSPlugin Output Order


I’ve use TAO 4.0 to train a model then converted it to a TensorRT engine. The output of the TensorRT engine that I’m interested in has a shape of [1, keepTopK, 7], as I understand it contains Box Class(1 value), Box Score(1 value) and bounding box location (4 values); Reference.

My question is, according to the output shape there is suppose to be 7 values, but only 6 values are explained, what about the other value? Also, what is the order of these values, Is there a documentation/reference that you can point me to?

Refer to TensorRT/plugin/nmsPlugin at release/8.6 · NVIDIA/TensorRT · GitHub

I looked at the link that you referenced, but it still doesn’t answer my question, it just vaguely mentions the existence of these values.
I’m looking for something like the following:

[0] → class_id ?
[1] → score ?
[2] → left_x ?
[3] → bottom_y ?
[4] → right_x or width ?
[5] → top_y or height?
[6] → ???

I need to know these mappings, so that I can properly evaluate and construct the proper bounding boxes…

You can refer to TensorRT/gatherTopDetections.cu at release/8.6 · NVIDIA/TensorRT · GitHub or deepstream_tao_apps/nvdsinfer_custombboxparser_tao.cpp at master · NVIDIA-AI-IOT/deepstream_tao_apps · GitHub

1 Like

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