The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation over boxes for object detection networks.
More...
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. |
◆ shareLocation
bool nvinfer1::plugin::NMSParameters::shareLocation |
◆ backgroundLabelId
int32_t nvinfer1::plugin::NMSParameters::backgroundLabelId |
◆ numClasses
int32_t nvinfer1::plugin::NMSParameters::numClasses |
◆ topK
int32_t nvinfer1::plugin::NMSParameters::topK |
◆ keepTopK
int32_t nvinfer1::plugin::NMSParameters::keepTopK |
◆ scoreThreshold
float nvinfer1::plugin::NMSParameters::scoreThreshold |
◆ iouThreshold
float nvinfer1::plugin::NMSParameters::iouThreshold |
◆ isNormalized
bool nvinfer1::plugin::NMSParameters::isNormalized |
The documentation for this struct was generated from the following file: