Go to the documentation of this file.
50 #ifndef NV_INFER_PLUGIN_H
51 #define NV_INFER_PLUGIN_H
79 float iouThreshold,
float minBoxSize,
float spatialScale,
nvinfer1::DimsHW pooling,
92 const nvinfer1::Weights* scales,
bool acrossSpatial,
bool channelShared,
float eps);
198 #endif // NV_INFER_PLUGIN_H
An array of weights used as a layer parameter.
Definition: NvInferRuntime.h:152
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
Definition: NvInferPluginUtils.h:229
Plugin class for user-implemented layers.
Definition: NvInferRuntimeCommon.h:414
nvinfer1::IPluginV2 * createNMSPlugin(nvinfer1::plugin::DetectionOutputParameters param)
The DetectionOutput plugin layer generates the detection output based on location and confidence pred...
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
nvinfer1::IPluginV2 * createPriorBoxPlugin(nvinfer1::plugin::PriorBoxParameters param)
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
Descriptor for two-dimensional spatial data.
Definition: NvInferLegacyDims.h:110
nvinfer1::IPluginV2 * createAnchorGeneratorPlugin(nvinfer1::plugin::GridAnchorParameters *param, int32_t numLayers)
The Grid Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratio...
bool initLibNvInferPlugins(void *logger, const char *libNamespace)
Initialize and register all the existing TensorRT plugins to the Plugin Registry with an optional nam...
nvinfer1::IPluginV2 * createInstanceNormalizationPlugin(float epsilon, nvinfer1::Weights scale_weights, nvinfer1::Weights bias_weights)
The Instance Normalization Plugin computes the instance normalization of an input tensor....
nvinfer1::IPluginV2 * createSplitPlugin(int32_t axis, int32_t *output_lengths, int32_t noutput)
The Split Plugin performs a split operation on the input tensor. It splits the input tensor into seve...
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....
nvinfer1::IPluginV2 * createRPNROIPlugin(int32_t featureStride, int32_t preNmsTop, int32_t 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....
nvinfer1::IPluginV2 * createBatchedNMSPlugin(nvinfer1::plugin::NMSParameters param)
The BatchedNMS Plugin performs non_max_suppression on the input boxes, per batch, across all classes....
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
nvinfer1::IPluginV2 * createRegionPlugin(nvinfer1::plugin::RegionParameters params)
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
nvinfer1::IPluginV2 * createReorgPlugin(int32_t stride)
The Reorg plugin reshapes input of shape CxHxW into a (C*stride*stride)x(H/stride)x(W/stride) shape,...
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