50 #ifndef NV_INFER_PLUGIN_H
51 #define NV_INFER_PLUGIN_H
113 virtual const char *
getName()
const = 0;
142 INvPlugin * createFasterRCNNPlugin(
int featureStride,
int preNmsTop,
143 int nmsMaxOut,
float iouThreshold,
float minBoxSize,
144 float spatialScale,
DimsHW pooling,
146 INvPlugin * createFasterRCNNPlugin(
const void * data,
size_t length);
156 INvPlugin * createSSDNormalizePlugin(
const Weights *scales,
bool acrossSpatial,
bool channelShared,
float eps);
157 INvPlugin * createSSDNormalizePlugin(
const void * data,
size_t length);
171 INvPlugin * createSSDPermutePlugin(
const void * data,
size_t length);
194 float *minSize, *maxSize, *aspectRatios;
195 int numMinSize, numMaxSize, numAspectRatios;
209 INvPlugin * createSSDPriorBoxPlugin(
const void * data,
size_t length);
235 bool shareLocation, varianceEncodedInTarget;
236 int backgroundLabelId, numClasses, topK, keepTopK;
237 float confidenceThreshold, nmsThreshold;
246 INvPlugin * createSSDDetectionOutputPlugin(
const void * data,
size_t length);
254 INvPlugin * createConcatPlugin(
int concatAxis,
bool ignoreBatch);
255 INvPlugin * createConcatPlugin(
const void * data,
size_t length);
263 INvPlugin * createPReLUPlugin(
float negSlope);
264 INvPlugin * createPReLUPlugin(
const void * data,
size_t length);
271 INvPlugin * createYOLOReorgPlugin(
int stride);
272 INvPlugin * createYOLOReorgPlugin(
const void * data,
size_t length);
305 INvPlugin * createYOLORegionPlugin(
const void * data,
size_t length);
309 #endif // NV_INFER_PLUGIN_H
Definition: NvInferPlugin.h:163
virtual const char * getName() const =0
get the name of the plugin from the ID
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:1713
Common interface for the NVidia created plugins.
Definition: NvInferPlugin.h:95
FasterRCNN fused plugin (RPN + ROI pooling)
SSD DetectionOutput plugin.
int EnumMax< PluginType >()
maximum number of elements in PluginType enum.
Definition: NvInferPlugin.h:82
Definition: NvInferPlugin.h:192
virtual void destroy()=0
destroy the plugin
an array of weights used as a layer parameter
Definition: NvInfer.h:381
Definition: NvInferPlugin.h:283
Definition: NvInferPlugin.h:296
descriptor for two-dimensional spatial data
Definition: NvInfer.h:156
Definition: NvInferPlugin.h:233