50 #ifndef NV_INFER_PLUGIN_H 51 #define NV_INFER_PLUGIN_H 54 #include "NvInferPluginUtils.h" 93 virtual const char*
getName()
const TRTNOEXCEPT = 0;
100 virtual void destroy() TRTNOEXCEPT = 0;
124 int nmsMaxOut,
float iouThreshold,
float minBoxSize,
125 float spatialScale,
DimsHW pooling,
226 int nmsMaxOut,
float iouThreshold,
float minBoxSize,
361 #endif // NV_INFER_PLUGIN_H Plugin class for user-implemented layers.
Definition: NvInferRuntimeCommon.h:344
The Permute plugin layer permutes the input tensor by changing the memory order of the data...
Definition: NvInferPluginUtils.h:97
TRT_DEPRECATED_API INvPlugin * createPReLUPlugin(float negSlope)
The PReLu plugin layer performs leaky ReLU for 4D tensors. Give an input value x, the PReLU layer com...
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...
virtual const char * getName() const =0
Get the name of the plugin from the ID.
nvinfer1::IPluginV2 * createPriorBoxPlugin(nvinfer1::plugin::PriorBoxParameters param)
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
PluginType
The type values for the various plugins.
Definition: NvInferPluginUtils.h:69
TRT_DEPRECATED_API 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...
virtual PluginType getPluginType() const =0
Get the parameter plugin ID.
TRT_DEPRECATED_API INvPlugin * createSSDNormalizePlugin(const Weights *scales, bool acrossSpatial, bool channelShared, float eps)
The Normalize plugin layer normalizes the input to have L2 norm of 1 with scale learnable.
Plugin class for user-implemented layers.
Definition: NvInferRuntime.h:134
#define TRT_DEPRECATED_API
Defines which symbols are exported.
Definition: NvInferRuntimeCommon.h:86
Common interface for the Nvidia created plugins.
Definition: NvInferPlugin.h:76
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...
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".
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...
The TensorRT API version 1 namespace.
nvinfer1::IPluginV2 * createSplitPlugin(int axis, int *output_lengths, int noutput)
The Split Plugin performs a split operation on the input tensor. It splits the input tensor into seve...
TRT_DEPRECATED_API INvPlugin * createSSDAnchorGeneratorPlugin(GridAnchorParameters *param, int numLayers)
The Grid Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratio...
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".
TRT_DEPRECATED_API nvinfer1::IPluginV2 * createLReLUPlugin(float negSlope)
The LReLu plugin layer performs leaky ReLU for 4D tensors. Give an input value x, the PReLU layer com...
The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all ...
Definition: NvInferPluginUtils.h:122
TRT_DEPRECATED_API INvPlugin * createConcatPlugin(int concatAxis, bool ignoreBatch)
The Concat plugin layer basically performs the concatention for 4D tensors. Unlike the Concatenation ...
bool initLibNvInferPlugins(void *logger, const char *libNamespace)
Initialize and register all the existing TensorRT plugins to the Plugin Registry with an optional nam...
The NMSParameters are used by the BatchedNMSPlugin for performing the non_max_suppression operation o...
Definition: NvInferPluginUtils.h:275
nvinfer1::IPluginV2 * createBatchedNMSPlugin(nvinfer1::plugin::NMSParameters param)
The BatchedNMS Plugin performs non_max_suppression on the input boxes, per batch, across all classes...
virtual void destroy()=0
Destroy the plugin.
An array of weights used as a layer parameter.
Definition: NvInferRuntime.h:98
TRT_DEPRECATED_API INvPlugin * createSSDDetectionOutputPlugin(DetectionOutputParameters param)
Definition: NvInferPluginUtils.h:250
TRT_DEPRECATED_API INvPlugin * createYOLOReorgPlugin(int stride)
The Reorg plugin layer maps the 512x26x26 feature map onto a 2048x13x13 feature map, so that it can be concatenated with the feature maps at 13x13 resolution.
nvinfer1::IPluginV2 * createNMSPlugin(nvinfer1::plugin::DetectionOutputParameters param)
The DetectionOutput plugin layer generates the detection output based on location and confidence pred...
nvinfer1::IPluginV2 * createRegionPlugin(nvinfer1::plugin::RegionParameters params)
The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for...
Descriptor for two-dimensional spatial data.
Definition: NvInfer.h:115
TRT_DEPRECATED_API INvPlugin * createSSDPriorBoxPlugin(PriorBoxParameters param)
The Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratios acr...
Definition: NvInferPluginUtils.h:175
The DetectionOutput plugin layer generates the detection output based on location and confidence pred...
Definition: NvInferPluginUtils.h:213
TRT_DEPRECATED_API INvPlugin * createFasterRCNNPlugin(int featureStride, int preNmsTop, int nmsMaxOut, float iouThreshold, float minBoxSize, float spatialScale, DimsHW pooling, Weights anchorRatios, Weights anchorScales)
Create a plugin layer that fuses the RPN and ROI pooling using user-defined parameters.
TRT_DEPRECATED_API INvPlugin * createSSDPermutePlugin(Quadruple permuteOrder)