TensorRT
8.0.3
|
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 |
int32_t | backgroundLabelId |
int32_t | numClasses |
int32_t | topK |
int32_t | 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. |
Copyright © 2024 NVIDIA Corporation
Privacy Policy |
Manage My Privacy |
Do Not Sell or Share My Data |
Terms of Service |
Accessibility |
Corporate Policies |
Product Security |
Contact