TensorRT 8.5.3
NvInferPluginUtils.h
Go to the documentation of this file.
1/*
2 * SPDX-FileCopyrightText: Copyright (c) 1993-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
4 *
5 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
6 * property and proprietary rights in and to this material, related
7 * documentation and any modifications thereto. Any use, reproduction,
8 * disclosure or distribution of this material and related documentation
9 * without an express license agreement from NVIDIA CORPORATION or
10 * its affiliates is strictly prohibited.
11 */
12
13#ifndef NV_INFER_PLUGIN_UTILS_H
14#define NV_INFER_PLUGIN_UTILS_H
15
17
24
25namespace nvinfer1
26{
27namespace plugin
28{
29
35typedef struct
36{
37 int32_t data[4];
38} Quadruple;
39
61{
64 bool flip;
65 bool clip;
66 float variance[4];
67 int32_t imgH, imgW;
68 float stepH, stepW;
69 float offset;
70};
71
88{
89 int32_t poolingH;
90 int32_t poolingW;
92 int32_t preNmsTop;
93 int32_t nmsMaxOut;
99};
100
101
115{
118 int32_t numAspectRatios, H, W;
119 float variance[4];
120};
121
126enum class CodeTypeSSD : int32_t
127{
128 CORNER = 0,
129 CENTER_SIZE = 1,
130 CORNER_SIZE = 2,
131 TF_CENTER = 3
132};
133
154{
159 int32_t inputOrder[3];
162 bool isBatchAgnostic{true};
163};
164
169{
170 int32_t* leaf;
171 int32_t n;
172 int32_t* parent;
173 int32_t* child;
174 int32_t* group;
175 char** name;
176
177 int32_t groups;
178 int32_t* groupSize;
179 int32_t* groupOffset;
180};
181
193{
194 int32_t num;
195 int32_t coords;
196 int32_t classes;
198};
199
216
218{
223};
224
225} // namespace plugin
226} // namespace nvinfer1
227
228#endif // NV_INFER_PLUGIN_UTILS_H
CodeTypeSSD
The type of encoding used for decoding the bounding boxes and loc_data.
Definition: NvInferPluginUtils.h:127
@ 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:154
int32_t backgroundLabelId
Definition: NvInferPluginUtils.h:156
float confidenceThreshold
Definition: NvInferPluginUtils.h:157
bool confSigmoid
Definition: NvInferPluginUtils.h:160
int32_t inputOrder[3]
Definition: NvInferPluginUtils.h:159
bool isBatchAgnostic
Definition: NvInferPluginUtils.h:162
int32_t keepTopK
Definition: NvInferPluginUtils.h:156
bool shareLocation
Definition: NvInferPluginUtils.h:155
int32_t numClasses
Definition: NvInferPluginUtils.h:156
bool varianceEncodedInTarget
Definition: NvInferPluginUtils.h:155
int32_t topK
Definition: NvInferPluginUtils.h:156
bool isNormalized
Definition: NvInferPluginUtils.h:161
float nmsThreshold
Definition: NvInferPluginUtils.h:157
CodeTypeSSD codeType
Definition: NvInferPluginUtils.h:158
The Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratios acr...
Definition: NvInferPluginUtils.h:115
int32_t W
Definition: NvInferPluginUtils.h:118
float * aspectRatios
Definition: NvInferPluginUtils.h:117
int32_t H
Definition: NvInferPluginUtils.h:118
float minSize
Definition: NvInferPluginUtils.h:116
int32_t numAspectRatios
Definition: NvInferPluginUtils.h:118
float variance[4]
Definition: NvInferPluginUtils.h:119
float maxSize
Definition: NvInferPluginUtils.h:116
The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation o...
Definition: NvInferPluginUtils.h:218
int32_t topK
Definition: NvInferPluginUtils.h:220
float iouThreshold
Definition: NvInferPluginUtils.h:221
int32_t numClasses
Definition: NvInferPluginUtils.h:220
int32_t backgroundLabelId
Definition: NvInferPluginUtils.h:220
int32_t keepTopK
Definition: NvInferPluginUtils.h:220
bool shareLocation
Definition: NvInferPluginUtils.h:219
float scoreThreshold
Definition: NvInferPluginUtils.h:221
bool isNormalized
Definition: NvInferPluginUtils.h:222
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
Definition: NvInferPluginUtils.h:61
float * maxSize
Definition: NvInferPluginUtils.h:62
float stepH
Definition: NvInferPluginUtils.h:68
float variance[4]
Definition: NvInferPluginUtils.h:66
float offset
Definition: NvInferPluginUtils.h:69
int32_t imgW
Definition: NvInferPluginUtils.h:67
int32_t numMaxSize
Definition: NvInferPluginUtils.h:63
bool clip
Definition: NvInferPluginUtils.h:65
int32_t numMinSize
Definition: NvInferPluginUtils.h:63
float stepW
Definition: NvInferPluginUtils.h:68
bool flip
Definition: NvInferPluginUtils.h:64
float * aspectRatios
Definition: NvInferPluginUtils.h:62
int32_t numAspectRatios
Definition: NvInferPluginUtils.h:63
int32_t imgH
Definition: NvInferPluginUtils.h:67
float * minSize
Definition: NvInferPluginUtils.h:62
The Permute plugin layer permutes the input tensor by changing the memory order of the data....
Definition: NvInferPluginUtils.h:36
RPROIParams is used to create the RPROIPlugin instance. It contains:
Definition: NvInferPluginUtils.h:88
int32_t featureStride
Definition: NvInferPluginUtils.h:91
int32_t poolingH
Definition: NvInferPluginUtils.h:89
int32_t preNmsTop
Definition: NvInferPluginUtils.h:92
int32_t anchorsScaleCount
Definition: NvInferPluginUtils.h:95
float spatialScale
Definition: NvInferPluginUtils.h:98
float iouThreshold
Definition: NvInferPluginUtils.h:96
int32_t nmsMaxOut
Definition: NvInferPluginUtils.h:93
int32_t anchorsRatioCount
Definition: NvInferPluginUtils.h:94
int32_t poolingW
Definition: NvInferPluginUtils.h:90
float minBoxSize
Definition: NvInferPluginUtils.h:97
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
Definition: NvInferPluginUtils.h:193
int32_t num
Definition: NvInferPluginUtils.h:194
softmaxTree * smTree
Definition: NvInferPluginUtils.h:197
int32_t coords
Definition: NvInferPluginUtils.h:195
int32_t classes
Definition: NvInferPluginUtils.h:196
When performing yolo9000, softmaxTree is helping to do softmax on confidence scores,...
Definition: NvInferPluginUtils.h:169
int32_t * parent
Definition: NvInferPluginUtils.h:172
int32_t * leaf
Definition: NvInferPluginUtils.h:170
int32_t * group
Definition: NvInferPluginUtils.h:174
int32_t * groupOffset
Definition: NvInferPluginUtils.h:179
int32_t n
Definition: NvInferPluginUtils.h:171
int32_t * child
Definition: NvInferPluginUtils.h:173
char ** name
Definition: NvInferPluginUtils.h:175
int32_t groups
Definition: NvInferPluginUtils.h:177
int32_t * groupSize
Definition: NvInferPluginUtils.h:178

  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