Go to the documentation of this file.
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;
265 #endif // NV_INFER_PLUGIN_UTILS_H
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
Definition: NvInferPluginUtils.h:229
The Permute plugin layer permutes the input tensor by changing the memory order of the data....
Definition: NvInferPluginUtils.h:72
The Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratios acr...
Definition: NvInferPluginUtils.h:151
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
Definition: NvInferPluginUtils.h:97
RPROIParams is used to create the RPROIPlugin instance. It contains:
Definition: NvInferPluginUtils.h:124
The TensorRT API version 1 namespace.
When performing yolo9000, softmaxTree is helping to do softmax on confidence scores,...
Definition: NvInferPluginUtils.h:205
CodeTypeSSD
The type of encoding used for decoding the bounding boxes and loc_data.
Definition: NvInferPluginUtils.h:163
The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation o...
Definition: NvInferPluginUtils.h:254
The DetectionOutput plugin layer generates the detection output based on location and confidence pred...
Definition: NvInferPluginUtils.h:190
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