NvInferPlugin.h File Reference

#include "NvInfer.h"

Go to the source code of this file.

Classes
class	nvinfer1::plugin::INvPlugin
	Common interface for the Nvidia created plugins. More...
struct	nvinfer1::plugin::Quadruple
	The Permute plugin layer permutes the input tensor by changing the memory order of the data. Quadruple defines a structure that contains an array of 4 integers. They can represent the permute orders or the strides in each dimension. More...
struct	nvinfer1::plugin::PriorBoxParameters
	The PriorBox plugin layer generates the prior boxes of designated sizes and aspect ratios across all dimensions . PriorBoxParameters defines a set of parameters for creating the PriorBox plugin layer. It contains: More...
struct	nvinfer1::plugin::GridAnchorParameters
	The Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratios across all dimensions . GridAnchorParameters defines a set of parameters for creating the plugin layer for all feature maps. It contains: More...
struct	nvinfer1::plugin::DetectionOutputParameters
	The DetectionOutput plugin layer generates the detection output based on location and confidence predictions by doing non maximum suppression. DetectionOutputParameters defines a set of parameters for creating the DetectionOutput plugin layer. It contains: More...
struct	nvinfer1::plugin::softmaxTree
	The Region plugin layer performs region proposal calculation: generate 5 bounding boxes per cell (for yolo9000, generate 3 bounding boxes per cell). For each box, calculating its probablities of objects detections from 80 pre-defined classifications (yolo9000 has 9416 pre-defined classifications, and these 9416 items are organized as work-tree structure). RegionParameters defines a set of parameters for creating the Region plugin layer. More...
struct	nvinfer1::plugin::RegionParameters

Namespaces
	nvinfer1
	The TensorRT API version 1 namespace.

Enumerations
enum	nvinfer1::PluginType : int {   nvinfer1::PluginType::kFASTERRCNN = 0, nvinfer1::PluginType::kNORMALIZE = 1, nvinfer1::PluginType::kPERMUTE = 2, nvinfer1::PluginType::kPRIORBOX = 3,   nvinfer1::PluginType::kSSDDETECTIONOUTPUT = 4, nvinfer1::PluginType::kCONCAT = 5, nvinfer1::PluginType::kPRELU = 6, nvinfer1::PluginType::kYOLOREORG = 7,   nvinfer1::PluginType::kYOLOREGION = 8, nvinfer1::PluginType::kANCHORGENERATOR = 9 }
	The type values for the various plugins. More...
enum	CodeTypeSSD : int { nvinfer1::plugin::CodeTypeSSD::CORNER = 0, nvinfer1::plugin::CodeTypeSSD::CENTER_SIZE = 1, nvinfer1::plugin::CodeTypeSSD::CORNER_SIZE = 2, nvinfer1::plugin::CodeTypeSSD::TF_CENTER = 3 }
	The type of encoding used for decoding the bounding boxes and loc_data.

Functions
template<>
int	nvinfer1::EnumMax< PluginType > ()
TENSORRTAPI INvPlugin *	nvinfer1::plugin::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. More...
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createFasterRCNNPlugin (const void *data, size_t length)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::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. More...
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createSSDNormalizePlugin (const void *data, size_t length)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createSSDPermutePlugin (Quadruple permuteOrder)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createSSDPermutePlugin (const void *data, size_t length)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createSSDPriorBoxPlugin (PriorBoxParameters param)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createSSDPriorBoxPlugin (const void *data, size_t length)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createSSDAnchorGeneratorPlugin (GridAnchorParameters *param, int numLayers)
	The Grid Anchor Generator plugin layer generates the prior boxes of designated sizes and aspect ratios across all dimensions for all feature maps. GridAnchorParameters defines a set of parameters for creating the GridAnchorGenerator plugin layer.
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createSSDAnchorGeneratorPlugin (const void *data, size_t length)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createSSDDetectionOutputPlugin (DetectionOutputParameters param)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createSSDDetectionOutputPlugin (const void *data, size_t length)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createConcatPlugin (int concatAxis, bool ignoreBatch)
	The Concat plugin layer basically performs the concatention for 4D tensors. Unlike the Concatenation layer in early version of TensorRT, it allows the user to specify the axis along which to concatenate. The axis can be 1 (across channel), 2 (across H), or 3 (across W). More particularly, this Concat plugin layer also implements the "ignoring the batch dimension" switch. If turned on, all the input tensors will be treated as if their batch sizes were 1. More...
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createConcatPlugin (const void *data, size_t length)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createPReLUPlugin (float negSlope)
	The PReLu plugin layer performs leaky ReLU for 4D tensors. Give an input value x, the PReLU layer computes the output as x if x > 0 and negative_slope //! x if x <= 0. More...
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createPReLUPlugin (const void *data, size_t length)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::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. More...
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createYOLOReorgPlugin (const void *data, size_t length)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createYOLORegionPlugin (RegionParameters params)
TENSORRTAPI INvPlugin *	nvinfer1::plugin::createYOLORegionPlugin (const void *data, size_t length)

Detailed Description

This is the API for the Nvidia provided TensorRT plugins.