BatchedNMSPlugin bounding box input format

[zhangjun]

Description

Hi, BatchedNMSPlugin plugin makes [ymin, xmin, ymax, xmax] as the default bounding box input format.

https://github.com/NVIDIA/TensorRT/blob/87f3394404ff9f9ec92c906cd4c39b5562aea42e/plugin/batchedNMSPlugin/batchedNMSInference.cu#L113-L119

How can I use the bounding box input format of [xmin, ymin, xmax, ymax]?

Can this parameter be configured?

Environment

TensorRT Version: 8.0.3.4 NVIDIA GPU: T4 NVIDIA Driver Version: 460.32.03 CUDA Version: 11.2 CUDNN Version: 8.2.1 Operating System: CentOS 7 Python Version (if applicable): 3.7 Tensorflow Version (if applicable): PyTorch Version (if applicable): Baremetal or Container (if so, version):

Relevant Files

Steps To Reproduce

[zerollzeng]

@kevinch-nv ^ ^

[triple-Mu]

Maybe you can only input xyxy format. EfficientNMS support both xyxy and xywh format by attrs bbox coding.

[ttyio]

closing since no activity for more than 14 days, please reopen if you still have question, thanks!