Class used for parsing Caffe models. More...

#include <NvCaffeParser.h>

Public Member Functions
virtual const IBlobNameToTensor *	parse (const char deploy, const char model, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightType)=0
	Parse a prototxt file and a binaryproto Caffe model to extract network configuration and weights associated with the network, respectively. More...

virtual IBinaryProtoBlob *	parseBinaryProto (const char *fileName)=0
	Parse and extract data stored in binaryproto file. More...

virtual void	setProtobufBufferSize (size_t size)=0
	Set buffer size for the parsing and storage of the learned model. More...

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...

virtual void	destroy ()=0
	Destroy this ICaffeParser object.

Detailed Description

Class used for parsing Caffe models.

Allows users to export models trained using Caffe to TRT.

Member Function Documentation

virtual const IBlobNameToTensor* nvcaffeparser1::ICaffeParser::parse	(	const char *	deploy,
		const char *	model,
		nvinfer1::INetworkDefinition &	network,
		nvinfer1::DataType	weightType
	)

pure virtual

Parse a prototxt file and a binaryproto Caffe model to extract network configuration and weights associated with the network, respectively.

Parameters

deploy	The plain text, prototxt file used to define the network configuration.
model	The binaryproto Caffe model that contains the weights associated with the network.
network	Network in which the CaffeParser will fill the layers.
weightType	The type to which the weights will transformed.

Returns: A pointer to an IBlobNameToTensor object that contains the extracted data.

See Also: nvcaffeparser1::IBlobNameToTensor

virtual IBinaryProtoBlob* nvcaffeparser1::ICaffeParser::parseBinaryProto ( const char * fileName )

pure virtual

Parse and extract data stored in binaryproto file.

The binaryproto file contains data stored in a binary blob. parseBinaryProto() converts it to an IBinaryProtoBlob object which gives the user access to the data and meta-data about data.

Parameters

fileName Path to file containing binary proto.

Returns: A pointer to an IBinaryProtoBlob object that contains the extracted data.

See Also: nvcaffeparser1::IBinaryProtoBlob

virtual void nvcaffeparser1::ICaffeParser::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 nvcaffeparser1::ICaffeParser::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.

virtual void nvcaffeparser1::ICaffeParser::setProtobufBufferSize ( size_t size )

pure virtual

Set buffer size for the parsing and storage of the learned model.

Parameters

size	The size of the buffer specified as the number of bytes.

Note: Default size is 2^30 bytes.

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

NvCaffeParser.h

Public Member Functions

Detailed Description

Member Function Documentation