|
virtual const IBlobNameToTensor * | parse (const char *deploy, const char *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 const IBlobNameToTensor * | parseBuffers (const uint8_t *deployBuffer, std::size_t deployLength, const uint8_t *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 (const char *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 (const char *libNamespace) noexcept=0 |
| Set the namespace used to lookup and create plugins in the network.
|
|
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.
- Warning
- Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.
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.
- Parameters
-
recorder | The error recorder to register with this interface. |
- See also
- getErrorRecorder()