50#ifndef NV_INFER_PLUGIN_UTILS_H
51#define NV_INFER_PLUGIN_UTILS_H
99 float *minSize, *maxSize, *aspectRatios;
100 int32_t numMinSize, numMaxSize, numAspectRatios;
128 int32_t featureStride;
131 int32_t anchorsRatioCount;
132 int32_t anchorsScaleCount;
153 float minSize, maxSize;
155 int32_t numAspectRatios, H, W;
192 bool shareLocation, varianceEncodedInTarget;
193 int32_t backgroundLabelId, numClasses, topK, keepTopK;
194 float confidenceThreshold, nmsThreshold;
196 int32_t inputOrder[3];
199 bool isBatchAgnostic{
true};
216 int32_t* groupOffset;
257 int32_t backgroundLabelId, numClasses, topK, keepTopK;
258 float scoreThreshold, iouThreshold;
CodeTypeSSD
The type of encoding used for decoding the bounding boxes and loc_data.
Definition: NvInferPluginUtils.h:164
@ CENTER_SIZE
Use box centers and size.
@ TF_CENTER
Use box centers and size but flip x and y coordinates.
@ CORNER_SIZE
Use box centers and size.
The TensorRT API version 1 namespace.
The DetectionOutput plugin layer generates the detection output based on location and confidence pred...
Definition: NvInferPluginUtils.h:191
The Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratios acr...
Definition: NvInferPluginUtils.h:152
The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation o...
Definition: NvInferPluginUtils.h:255
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
Definition: NvInferPluginUtils.h:98
The Permute plugin layer permutes the input tensor by changing the memory order of the data....
Definition: NvInferPluginUtils.h:73
RPROIParams is used to create the RPROIPlugin instance. It contains:
Definition: NvInferPluginUtils.h:125
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
Definition: NvInferPluginUtils.h:230
When performing yolo9000, softmaxTree is helping to do softmax on confidence scores,...
Definition: NvInferPluginUtils.h:206