TensorRT
7.0.0.11
|
The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation over boxes for object detection networks. More...
#include <NvInferPluginUtils.h>
Public Attributes | |
bool | shareLocation |
int | backgroundLabelId |
int | numClasses |
int | topK |
int | keepTopK |
float | scoreThreshold |
float | iouThreshold |
bool | isNormalized |
The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation over boxes for object detection networks.
shareLocation | If set to true, the boxes inputs are shared across all classes. If set to false, the boxes input should account for per class box data. |
backgroundLabelId | Label ID for the background class. If there is no background class, set it as -1 |
numClasses | Number of classes in the network. |
topK | Number of bounding boxes to be fed into the NMS step. |
keepTopK | Number of total bounding boxes to be kept per image after NMS step. Should be less than or equal to the topK value. |
scoreThreshold | Scalar threshold for score (low scoring boxes are removed). |
iouThreshold | scalar threshold for IOU (new boxes that have high IOU overlap with previously selected boxes are removed). |
isNormalized | Set to false, if the box coordinates are not normalized, i.e. not in the range [0,1]. Defaults to false. |