TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
nvinfer1::plugin::DetectionOutputParameters Struct Reference

The DetectionOutput plugin layer generates the detection output based on location and confidence predictions by doing non maximum suppression. More...

Collaboration diagram for nvinfer1::plugin::DetectionOutputParameters:

Public Attributes

bool shareLocation
 
bool varianceEncodedInTarget
 
int32_t backgroundLabelId
 
int32_t numClasses
 
int32_t topK
 
int32_t keepTopK
 
float confidenceThreshold
 
float nmsThreshold
 
CodeTypeSSD codeType
 
int32_t inputOrder [3]
 
bool confSigmoid
 
bool isNormalized
 

Detailed Description

The DetectionOutput plugin layer generates the detection output based on location and confidence predictions by doing non maximum suppression.

This plugin first decodes the bounding boxes based on the anchors generated. It then performs non_max_suppression on the decoded bouding boxes. DetectionOutputParameters defines a set of parameters for creating the DetectionOutput plugin layer. It contains:

Parameters
shareLocationIf true, bounding box are shared among different classes.
varianceEncodedInTargetIf true, variance is encoded in target. Otherwise we need to adjust the predicted offset accordingly.
backgroundLabelIdBackground label ID. If there is no background class, set it as -1.
numClassesNumber of classes to be predicted.
topKNumber of boxes per image with top confidence scores that are fed into the NMS algorithm.
keepTopKNumber of total bounding boxes to be kept per image after NMS step.
confidenceThresholdOnly consider detections whose confidences are larger than a threshold.
nmsThresholdThreshold to be used in NMS.
codeTypeType of coding method for bbox.
inputOrderSpecifies the order of inputs {loc_data, conf_data, priorbox_data}.
confSigmoidSet to true to calculate sigmoid of confidence scores.
isNormalizedSet to true if bounding box data is normalized by the network.

Member Data Documentation

◆ shareLocation

bool nvinfer1::plugin::DetectionOutputParameters::shareLocation

◆ varianceEncodedInTarget

bool nvinfer1::plugin::DetectionOutputParameters::varianceEncodedInTarget

◆ backgroundLabelId

int32_t nvinfer1::plugin::DetectionOutputParameters::backgroundLabelId

◆ numClasses

int32_t nvinfer1::plugin::DetectionOutputParameters::numClasses

◆ topK

int32_t nvinfer1::plugin::DetectionOutputParameters::topK

◆ keepTopK

int32_t nvinfer1::plugin::DetectionOutputParameters::keepTopK

◆ confidenceThreshold

float nvinfer1::plugin::DetectionOutputParameters::confidenceThreshold

◆ nmsThreshold

float nvinfer1::plugin::DetectionOutputParameters::nmsThreshold

◆ codeType

CodeTypeSSD nvinfer1::plugin::DetectionOutputParameters::codeType

◆ inputOrder

int32_t nvinfer1::plugin::DetectionOutputParameters::inputOrder[3]

◆ confSigmoid

bool nvinfer1::plugin::DetectionOutputParameters::confSigmoid

◆ isNormalized

bool nvinfer1::plugin::DetectionOutputParameters::isNormalized

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