TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
nvinfer1::plugin::DetectionLayerPluginCreator Class Reference
Inheritance diagram for nvinfer1::plugin::DetectionLayerPluginCreator:
Collaboration diagram for nvinfer1::plugin::DetectionLayerPluginCreator:

Public Member Functions

 DetectionLayerPluginCreator ()
 
 ~DetectionLayerPluginCreator ()
 
const char * getPluginName () const override
 Return the plugin name. More...
 
const char * getPluginVersion () const override
 Return the plugin version. More...
 
const PluginFieldCollectiongetFieldNames () override
 Return a list of fields that needs to be passed to createPlugin. More...
 
IPluginV2ExtcreatePlugin (const char *name, const PluginFieldCollection *fc) override
 Return a plugin object. More...
 
IPluginV2ExtdeserializePlugin (const char *name, const void *data, size_t length) override
 Called during deserialization of plugin layer. More...
 
void setPluginNamespace (const char *libNamespace) override
 Set the namespace of the plugin creator based on the plugin library it belongs to. More...
 
const char * getPluginNamespace () const override
 Return the namespace of the plugin creator object. More...
 
virtual int32_t getTensorRTVersion () const
 Return the version of the API the plugin creator was compiled with. More...
 

Protected Attributes

std::string mNamespace
 

Private Attributes

int mNbClasses
 
int mKeepTopK
 
float mScoreThreshold
 
float mIOUThreshold
 

Static Private Attributes

static PluginFieldCollection mFC {}
 
static std::vector< PluginFieldmPluginAttributes
 

Constructor & Destructor Documentation

◆ DetectionLayerPluginCreator()

DetectionLayerPluginCreator::DetectionLayerPluginCreator ( )

◆ ~DetectionLayerPluginCreator()

nvinfer1::plugin::DetectionLayerPluginCreator::~DetectionLayerPluginCreator ( )
inline

Member Function Documentation

◆ getPluginName()

const char * DetectionLayerPluginCreator::getPluginName ( ) const
overridevirtual

Return the plugin name.

Implements nvinfer1::IPluginCreator.

◆ getPluginVersion()

const char * DetectionLayerPluginCreator::getPluginVersion ( ) const
overridevirtual

Return the plugin version.

Implements nvinfer1::IPluginCreator.

◆ getFieldNames()

const PluginFieldCollection * DetectionLayerPluginCreator::getFieldNames ( )
overridevirtual

Return a list of fields that needs to be passed to createPlugin.

See also
PluginFieldCollection

Implements nvinfer1::IPluginCreator.

◆ createPlugin()

IPluginV2Ext * DetectionLayerPluginCreator::createPlugin ( const char *  name,
const PluginFieldCollection fc 
)
overridevirtual

Return a plugin object.

Return nullptr in case of error.

Implements nvinfer1::IPluginCreator.

◆ deserializePlugin()

IPluginV2Ext * DetectionLayerPluginCreator::deserializePlugin ( const char *  name,
const void *  serialData,
size_t  serialLength 
)
overridevirtual

Called during deserialization of plugin layer.

Return a plugin object.

Implements nvinfer1::IPluginCreator.

◆ setPluginNamespace()

void nvinfer1::plugin::BaseCreator::setPluginNamespace ( const char *  pluginNamespace)
inlineoverridevirtualinherited

Set the namespace of the plugin creator based on the plugin library it belongs to.

This can be set while registering the plugin creator.

See also
IPluginRegistry::registerCreator()

Implements nvinfer1::IPluginCreator.

◆ getPluginNamespace()

const char* nvinfer1::plugin::BaseCreator::getPluginNamespace ( ) const
inlineoverridevirtualinherited

Return the namespace of the plugin creator object.

Implements nvinfer1::IPluginCreator.

◆ getTensorRTVersion()

virtual int32_t nvinfer1::IPluginCreator::getTensorRTVersion ( ) const
inlinevirtualinherited

Return the version of the API the plugin creator was compiled with.

Member Data Documentation

◆ mFC

PluginFieldCollection DetectionLayerPluginCreator::mFC {}
staticprivate

◆ mNbClasses

int nvinfer1::plugin::DetectionLayerPluginCreator::mNbClasses
private

◆ mKeepTopK

int nvinfer1::plugin::DetectionLayerPluginCreator::mKeepTopK
private

◆ mScoreThreshold

float nvinfer1::plugin::DetectionLayerPluginCreator::mScoreThreshold
private

◆ mIOUThreshold

float nvinfer1::plugin::DetectionLayerPluginCreator::mIOUThreshold
private

◆ mPluginAttributes

std::vector< PluginField > DetectionLayerPluginCreator::mPluginAttributes
staticprivate

◆ mNamespace

std::string nvinfer1::plugin::BaseCreator::mNamespace
protectedinherited

The documentation for this class was generated from the following files: