50 #ifndef NV_INFER_PLUGIN_H
51 #define NV_INFER_PLUGIN_H
86 inline int EnumMax<PluginType>()
120 virtual const char*
getName()
const = 0;
150 TENSORRTAPI
INvPlugin* createFasterRCNNPlugin(
int featureStride,
int preNmsTop,
151 int nmsMaxOut,
float iouThreshold,
float minBoxSize,
152 float spatialScale,
DimsHW pooling,
154 TENSORRTAPI
INvPlugin* createFasterRCNNPlugin(
const void* data,
size_t length);
164 TENSORRTAPI
INvPlugin* createSSDNormalizePlugin(
const Weights* scales,
bool acrossSpatial,
bool channelShared,
float eps);
165 TENSORRTAPI
INvPlugin* createSSDNormalizePlugin(
const void* data,
size_t length);
181 TENSORRTAPI
INvPlugin* createSSDPermutePlugin(
const void* data,
size_t length);
204 float *minSize, *maxSize, *aspectRatios;
205 int numMinSize, numMaxSize, numAspectRatios;
228 float minSize, maxSize;
230 int numAspectRatios, H, W;
239 TENSORRTAPI
INvPlugin* createSSDPriorBoxPlugin(
const void* data,
size_t length);
248 TENSORRTAPI
INvPlugin* createSSDAnchorGeneratorPlugin(
const void* data,
size_t length);
254 enum class CodeTypeSSD : int
282 bool shareLocation, varianceEncodedInTarget;
283 int backgroundLabelId, numClasses, topK, keepTopK;
284 float confidenceThreshold, nmsThreshold;
285 CodeTypeSSD codeType;
296 TENSORRTAPI
INvPlugin* createSSDDetectionOutputPlugin(
const void* data,
size_t length);
306 TENSORRTAPI
INvPlugin* createConcatPlugin(
int concatAxis,
bool ignoreBatch);
307 TENSORRTAPI
INvPlugin* createConcatPlugin(
const void* data,
size_t length);
315 TENSORRTAPI
INvPlugin* createPReLUPlugin(
float negSlope);
316 TENSORRTAPI
INvPlugin* createPReLUPlugin(
const void* data,
size_t length);
323 TENSORRTAPI
INvPlugin* createYOLOReorgPlugin(
int stride);
324 TENSORRTAPI
INvPlugin* createYOLOReorgPlugin(
const void* data,
size_t length);
360 TENSORRTAPI
INvPlugin* createYOLORegionPlugin(
const void* data,
size_t length);
382 int backgroundLabelId, numClasses, topK, keepTopK;
383 float scoreThreshold, iouThreshold;
408 int nmsMaxOut,
float iouThreshold,
float minBoxSize,
518 #endif // NV_INFER_PLUGIN_H
TENSORRTAPI bool initLibNvInferPlugins(void *logger, const char *libNamespace)
Initialize and register all the existing TensorRT plugins to the Plugin Registry with an optional nam...
Plugin class for user-implemented layers.
Definition: NvInfer.h:3016
The Permute plugin layer permutes the input tensor by changing the memory order of the data...
Definition: NvInferPlugin.h:171
TENSORRTAPI nvinfer1::IPluginV2 * createRegionPlugin(nvinfer1::plugin::RegionParameters params)
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
virtual const char * getName() const =0
Get the name of the plugin from the ID.
SSD Grid Anchor Generator.
PluginType
The type values for the various plugins.
Definition: NvInferPlugin.h:70
virtual PluginType getPluginType() const =0
Get the parameter plugin ID.
Plugin class for user-implemented layers.
Definition: NvInfer.h:2849
Common interface for the Nvidia created plugins.
Definition: NvInferPlugin.h:103
TENSORRTAPI nvinfer1::IPluginV2 * createReorgPlugin(int stride)
The Reorg plugin reshapes input of shape CxHxW into a (C*stride*stride)x(H/stride)x(W/stride) shape...
FasterRCNN fused plugin (RPN + ROI pooling).
SSD DetectionOutput plugin.
TENSORRTAPI nvinfer1::IPluginV2 * createBatchedNMSPlugin(nvinfer1::plugin::NMSParameters param)
The BatchedNMS Plugin performs non_max_suppression on the input boxes, per batch, across all classes...
TENSORRTAPI nvinfer1::IPluginV2 * createRPNROIPlugin(int featureStride, int preNmsTop, int nmsMaxOut, float iouThreshold, float minBoxSize, float spatialScale, nvinfer1::DimsHW pooling, nvinfer1::Weights anchorRatios, nvinfer1::Weights anchorScales)
Create a plugin layer that fuses the RPN and ROI pooling using user-defined parameters. Registered plugin type "RPROI_TRT". Registered plugin version "1".
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
Definition: NvInferPlugin.h:202
The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation o...
Definition: NvInferPlugin.h:379
virtual void destroy()=0
Destroy the plugin.
An array of weights used as a layer parameter.
Definition: NvInfer.h:516
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
Definition: NvInferPlugin.h:337
TENSORRTAPI nvinfer1::IPluginV2 * createNMSPlugin(nvinfer1::plugin::DetectionOutputParameters param)
The DetectionOutput plugin layer generates the detection output based on location and confidence pred...
Definition: NvInferPlugin.h:351
TENSORRTAPI nvinfer1::IPluginV2 * createPriorBoxPlugin(nvinfer1::plugin::PriorBoxParameters param)
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
TENSORRTAPI nvinfer1::IPluginV2 * createNormalizePlugin(const nvinfer1::Weights *scales, bool acrossSpatial, bool channelShared, float eps)
The Normalize plugin layer normalizes the input to have L2 norm of 1 with scale learnable. Registered plugin type "Normalize_TRT". Registered plugin version "1".
Descriptor for two-dimensional spatial data.
Definition: NvInfer.h:224
TENSORRTAPI nvinfer1::IPluginV2 * createLReLUPlugin(float negSlope)
The LReLu plugin layer performs leaky ReLU for 4D tensors. Give an input value x, the PReLU layer com...
TENSORRTAPI nvinfer1::IPluginV2 * createClipPlugin(const char *layerName, float clipMin, float clipMax)
The Clip Plugin performs a clip operation on the input tensor. It clips the tensor values to a specif...
TENSORRTAPI nvinfer1::IPluginV2 * createAnchorGeneratorPlugin(nvinfer1::plugin::GridAnchorParameters *param, int numLayers)
The Grid Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratio...
The Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratios acr...
Definition: NvInferPlugin.h:226
The DetectionOutput plugin layer generates the detection output based on location and confidence pred...
Definition: NvInferPlugin.h:280