Class used for parsing models described using the UFF format. More...

Public Member Functions
virtual bool	registerInput (const char *inputName, nvinfer1::Dims inputDims, UffInputOrder inputOrder)=0
	Register an input name of a UFF network with the associated Dimensions. More...

virtual bool	registerOutput (const char *outputName)=0
	Register an output name of a UFF network. More...

virtual bool	parse (const char *file, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightsType=nvinfer1::DataType::kFLOAT)=0
	Parse a UFF file. More...

virtual bool	parseBuffer (const char *buffer, std::size_t size, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightsType=nvinfer1::DataType::kFLOAT)=0
	Parse a UFF buffer, useful if the file already live in memory. More...

virtual void	destroy ()=0

virtual int	getUffRequiredVersionMajor ()=0
	Return Version Major of the UFF.

virtual int	getUffRequiredVersionMinor ()=0
	Return Version Minor of the UFF.

virtual int	getUffRequiredVersionPatch ()=0
	Return Patch Version of the UFF.

virtual void	setPluginFactory (IPluginFactory *factory)=0
	Set the IPluginFactory used to create the user defined plugins. More...

virtual void	setPluginFactoryExt (IPluginFactoryExt *factory)=0
	Set the IPluginFactoryExt used to create the user defined pluginExts. More...

Detailed Description

Class used for parsing models described using the UFF format.

Member Function Documentation

virtual bool nvuffparser::IUffParser::parse	(	const char *	file,
		nvinfer1::INetworkDefinition &	network,
		nvinfer1::DataType	weightsType = `nvinfer1::DataType::kFLOAT`
	)

pure virtual

Parse a UFF file.

Parameters

file	File name of the UFF file.
network	Network in which the UFFParser will fill the layers.
weightsType	The type on which the weights will transformed in.

virtual bool nvuffparser::IUffParser::parseBuffer	(	const char *	buffer,
		std::size_t	size,
		nvinfer1::INetworkDefinition &	network,
		nvinfer1::DataType	weightsType = `nvinfer1::DataType::kFLOAT`
	)

pure virtual

Parse a UFF buffer, useful if the file already live in memory.

Parameters

buffer	Buffer of the UFF file.
size	Size of buffer of the UFF file.
network	Network in which the UFFParser will fill the layers.
weightsType	The type on which the weights will transformed in.

virtual bool nvuffparser::IUffParser::registerInput	(	const char *	inputName,
		nvinfer1::Dims	inputDims,
		UffInputOrder	inputOrder
	)

pure virtual

Register an input name of a UFF network with the associated Dimensions.

Parameters

inputName	Input name.
inputDims	Input Dimensions, always provide your dimensions in CHW even if your network input was in HWC in yout original framework.
inputOrder	Input order on which the framework input was originally.

virtual bool nvuffparser::IUffParser::registerOutput ( const char * outputName )

pure virtual

Register an output name of a UFF network.

Parameters

outputName Output name.

virtual void nvuffparser::IUffParser::setPluginFactory ( IPluginFactory * factory )

pure virtual

Set the IPluginFactory used to create the user defined plugins.

Parameters

factory Pointer to an instance of the user implmentation of IPluginFactory.

virtual void nvuffparser::IUffParser::setPluginFactoryExt ( IPluginFactoryExt * factory )

pure virtual

Set the IPluginFactoryExt used to create the user defined pluginExts.

Parameters

factory Pointer to an instance of the user implmentation of IPluginFactoryExt.

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