TensorRT
|
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. | |
Class used for parsing Caffe models.
Allows users to export models trained using Caffe to TRT.
|
pure virtual |
Parse a prototxt file and a binaryproto Caffe model to extract network configuration and weights associated with the network, respectively.
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. |
|
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.
fileName | Path to file containing binary proto. |
|
pure virtual |
Set the IPluginFactory used to create the user defined plugins.
factory | Pointer to an instance of the user implmentation of IPluginFactory. |
|
pure virtual |
Set the IPluginFactoryExt used to create the user defined pluginExts.
factory | Pointer to an instance of the user implmentation of IPluginFactoryExt. |
|
pure virtual |
Set buffer size for the parsing and storage of the learned model.
size | The size of the buffer specified as the number of bytes. |