The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation over boxes for object detection networks. More...

Detailed Description

The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation over boxes for object detection networks.

Parameters

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.

Member Data Documentation

int32_t nvinfer1::plugin::NMSParameters::backgroundLabelId

float nvinfer1::plugin::NMSParameters::iouThreshold

bool nvinfer1::plugin::NMSParameters::isNormalized

int32_t nvinfer1::plugin::NMSParameters::keepTopK

int32_t nvinfer1::plugin::NMSParameters::numClasses

float nvinfer1::plugin::NMSParameters::scoreThreshold

bool nvinfer1::plugin::NMSParameters::shareLocation

int32_t nvinfer1::plugin::NMSParameters::topK

The documentation for this struct was generated from the following file: