TensorRT 8.4.3
|
Class used for parsing Caffe models. More...
#include <NvCaffeParser.h>
Public Member Functions | |
virtual IBlobNameToTensor const * | parse (char const *deploy, char const *model, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightType) noexcept=0 |
Parse a prototxt file and a binaryproto Caffe model to extract network definition and weights associated with the network, respectively. More... | |
virtual IBlobNameToTensor const * | parseBuffers (uint8_t const *deployBuffer, std::size_t deployLength, uint8_t const *modelBuffer, std::size_t modelLength, nvinfer1::INetworkDefinition &network, nvinfer1::DataType weightType) noexcept=0 |
Parse a deploy prototxt and a binaryproto Caffe model from memory buffers to extract network definition and weights associated with the network, respectively. More... | |
virtual IBinaryProtoBlob * | parseBinaryProto (char const *fileName) noexcept=0 |
Parse and extract data stored in binaryproto file. More... | |
virtual void | setProtobufBufferSize (size_t size) noexcept=0 |
Set buffer size for the parsing and storage of the learned model. More... | |
virtual TRT_DEPRECATED void | destroy () noexcept=0 |
Destroy this ICaffeParser object. More... | |
virtual void | setPluginFactoryV2 (IPluginFactoryV2 *factory) noexcept=0 |
Set the IPluginFactoryV2 used to create the user defined pluginV2 objects. More... | |
virtual void | setPluginNamespace (char const *libNamespace) noexcept=0 |
Set the namespace used to lookup and create plugins in the network. More... | |
virtual | ~ICaffeParser () noexcept=default |
virtual void | setErrorRecorder (nvinfer1::IErrorRecorder *recorder) noexcept=0 |
Set the ErrorRecorder for this interface. More... | |
virtual nvinfer1::IErrorRecorder * | getErrorRecorder () const noexcept=0 |
get the ErrorRecorder assigned to this interface. More... | |
Class used for parsing Caffe models.
Allows users to export models trained using Caffe to TRT.
|
virtualdefaultnoexcept |
|
pure virtualnoexcept |
Destroy this ICaffeParser object.
delete
instead. Deprecated in TensorRT 8.0.
|
pure virtualnoexcept |
get the ErrorRecorder assigned to this interface.
Retrieves the assigned error recorder object for the given class. A nullptr will be returned if setErrorRecorder has not been called.
|
pure virtualnoexcept |
Parse a prototxt file and a binaryproto Caffe model to extract network definition and weights associated with the network, respectively.
deploy | The plain text, prototxt file used to define the network definition. |
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 virtualnoexcept |
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 virtualnoexcept |
Parse a deploy prototxt and a binaryproto Caffe model from memory buffers to extract network definition and weights associated with the network, respectively.
deployBuffer | The plain text deploy prototxt used to define the network definition. |
deployLength | The length of the deploy buffer. |
modelBuffer | The binaryproto Caffe memory buffer that contains the weights associated with the network. |
modelLength | The length of the model buffer. |
network | Network in which the CaffeParser will fill the layers. |
weightType | The type to which the weights will transformed. |
|
pure virtualnoexcept |
Set the ErrorRecorder for this interface.
Assigns the ErrorRecorder to this interface. The ErrorRecorder will track all errors during execution. This function will call incRefCount of the registered ErrorRecorder at least once. Setting recorder to nullptr unregisters the recorder with the interface, resulting in a call to decRefCount if a recorder has been registered.
If an error recorder is not set, messages will be sent to the global log stream.
recorder | The error recorder to register with this interface. |
|
pure virtualnoexcept |
Set the IPluginFactoryV2 used to create the user defined pluginV2 objects.
factory | Pointer to an instance of the user implementation of IPluginFactoryV2. |
|
pure virtualnoexcept |
Set the namespace used to lookup and create plugins in the network.
|
pure virtualnoexcept |
Set buffer size for the parsing and storage of the learned model.
size | The size of the buffer specified as the number of bytes. |
Copyright © 2024 NVIDIA Corporation
Privacy Policy |
Manage My Privacy |
Do Not Sell or Share My Data |
Terms of Service |
Accessibility |
Corporate Policies |
Product Security |
Contact