org.bytedeco.tensorrt.nvparsers

Class ICaffeParser

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.tensorrt.nvparsers.ICaffeParser

All Implemented Interfaces:

AutoCloseable

@Namespace(value="nvcaffeparser1")
 @Properties(inherit=nvparsers.class)
public class ICaffeParser
extends Pointer

\class ICaffeParser \brief 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.

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
ICaffeParser(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
void	destroy() Deprecated. Deprecated in TensorRT 8.0. Superseded by delete. \warning Calling destroy on a managed pointer will result in a double-free error.
IErrorRecorder	getErrorRecorder() \brief get the ErrorRecorder assigned to this interface.
IBlobNameToTensor	parse(BytePointer deploy, BytePointer model, INetworkDefinition network, int weightType)
IBlobNameToTensor	parse(String deploy, String model, INetworkDefinition network, nvinfer.DataType weightType) \brief Parse a prototxt file and a binaryproto Caffe model to extract network definition and weights associated with the network, respectively.
IBinaryProtoBlob	parseBinaryProto(BytePointer fileName)
IBinaryProtoBlob	parseBinaryProto(String fileName) \brief Parse and extract data stored in binaryproto file.
IBlobNameToTensor	parseBuffers(byte[] deployBuffer, long deployLength, byte[] modelBuffer, long modelLength, INetworkDefinition network, int weightType)
IBlobNameToTensor	parseBuffers(byte[] deployBuffer, long deployLength, byte[] modelBuffer, long modelLength, INetworkDefinition network, nvinfer.DataType weightType)
IBlobNameToTensor	parseBuffers(ByteBuffer deployBuffer, long deployLength, ByteBuffer modelBuffer, long modelLength, INetworkDefinition network, int weightType)
IBlobNameToTensor	parseBuffers(ByteBuffer deployBuffer, long deployLength, ByteBuffer modelBuffer, long modelLength, INetworkDefinition network, nvinfer.DataType weightType)
IBlobNameToTensor	parseBuffers(BytePointer deployBuffer, long deployLength, BytePointer modelBuffer, long modelLength, INetworkDefinition network, int weightType)
IBlobNameToTensor	parseBuffers(BytePointer deployBuffer, long deployLength, BytePointer modelBuffer, long modelLength, INetworkDefinition network, nvinfer.DataType weightType) \brief Parse a deploy prototxt and a binaryproto Caffe model from memory buffers to extract network definition and weights associated with the network, respectively.
void	setErrorRecorder(IErrorRecorder recorder) \brief Set the ErrorRecorder for this interface Assigns the ErrorRecorder to this interface.
void	setPluginFactoryV2(IPluginFactoryV2 factory) \brief Set the IPluginFactoryV2 used to create the user defined pluginV2 objects.
void	setPluginNamespace(BytePointer libNamespace)
void	setPluginNamespace(String libNamespace) \brief Set the namespace used to lookup and create plugins in the network.
void	setProtobufBufferSize(long size) \brief Set buffer size for the parsing and storage of the learned model.

Methods inherited from class org.bytedeco.javacpp.Pointer

address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

ICaffeParser

public ICaffeParser(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

Method Detail

parse

@Const
 @NoException(value=true)
public IBlobNameToTensor parse(String deploy,
                                                                String model,
                                                                @ByRef
                                                                INetworkDefinition network,
                                                                nvinfer.DataType weightType)

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

Parameters:

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.

Returns:

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

See Also:

nvcaffeparser1::IBlobNameToTensor

parse

@Const
 @NoException(value=true)
public IBlobNameToTensor parse(@Cast(value="const char*")
                                                                BytePointer deploy,
                                                                @Cast(value="const char*")
                                                                BytePointer model,
                                                                @ByRef
                                                                INetworkDefinition network,
                                                                @Cast(value="nvinfer1::DataType")
                                                                int weightType)

parseBuffers

@Const
 @NoException(value=true)
public IBlobNameToTensor parseBuffers(@Cast(value="const uint8_t*")
                                                                       BytePointer deployBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long deployLength,
                                                                       @Cast(value="const uint8_t*")
                                                                       BytePointer modelBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long modelLength,
                                                                       @ByRef
                                                                       INetworkDefinition network,
                                                                       nvinfer.DataType weightType)

\brief Parse a deploy prototxt and a binaryproto Caffe model from memory buffers to extract network definition and weights associated with the network, respectively.

Parameters:

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.

Returns:

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

See Also:

nvcaffeparser1::IBlobNameToTensor

parseBuffers

@Const
 @NoException(value=true)
public IBlobNameToTensor parseBuffers(@Cast(value="const uint8_t*")
                                                                       ByteBuffer deployBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long deployLength,
                                                                       @Cast(value="const uint8_t*")
                                                                       ByteBuffer modelBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long modelLength,
                                                                       @ByRef
                                                                       INetworkDefinition network,
                                                                       @Cast(value="nvinfer1::DataType")
                                                                       int weightType)

parseBuffers

@Const
 @NoException(value=true)
public IBlobNameToTensor parseBuffers(@Cast(value="const uint8_t*")
                                                                       byte[] deployBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long deployLength,
                                                                       @Cast(value="const uint8_t*")
                                                                       byte[] modelBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long modelLength,
                                                                       @ByRef
                                                                       INetworkDefinition network,
                                                                       nvinfer.DataType weightType)

parseBuffers

@Const
 @NoException(value=true)
public IBlobNameToTensor parseBuffers(@Cast(value="const uint8_t*")
                                                                       BytePointer deployBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long deployLength,
                                                                       @Cast(value="const uint8_t*")
                                                                       BytePointer modelBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long modelLength,
                                                                       @ByRef
                                                                       INetworkDefinition network,
                                                                       @Cast(value="nvinfer1::DataType")
                                                                       int weightType)

parseBuffers

@Const
 @NoException(value=true)
public IBlobNameToTensor parseBuffers(@Cast(value="const uint8_t*")
                                                                       ByteBuffer deployBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long deployLength,
                                                                       @Cast(value="const uint8_t*")
                                                                       ByteBuffer modelBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long modelLength,
                                                                       @ByRef
                                                                       INetworkDefinition network,
                                                                       nvinfer.DataType weightType)

parseBuffers

@Const
 @NoException(value=true)
public IBlobNameToTensor parseBuffers(@Cast(value="const uint8_t*")
                                                                       byte[] deployBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long deployLength,
                                                                       @Cast(value="const uint8_t*")
                                                                       byte[] modelBuffer,
                                                                       @Cast(value="std::size_t")
                                                                       long modelLength,
                                                                       @ByRef
                                                                       INetworkDefinition network,
                                                                       @Cast(value="nvinfer1::DataType")
                                                                       int weightType)

parseBinaryProto

@NoException(value=true)
public IBinaryProtoBlob parseBinaryProto(String fileName)

\brief 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

parseBinaryProto

@NoException(value=true)
public IBinaryProtoBlob parseBinaryProto(@Cast(value="const char*")
                                                                  BytePointer fileName)

setProtobufBufferSize

@NoException(value=true)
public void setProtobufBufferSize(@Cast(value="size_t")
                                                           long size)

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

destroy

@Deprecated
 @NoException(value=true)
public void destroy()

Deprecated. Deprecated in TensorRT 8.0. Superseded by delete. \warning Calling destroy on a managed pointer will result in a double-free error.

\brief Destroy this ICaffeParser object.

setPluginFactoryV2

@NoException(value=true)
public void setPluginFactoryV2(IPluginFactoryV2 factory)

\brief Set the IPluginFactoryV2 used to create the user defined pluginV2 objects.

Parameters:

factory - Pointer to an instance of the user implementation of IPluginFactoryV2.

setPluginNamespace

@NoException(value=true)
public void setPluginNamespace(String libNamespace)

\brief Set the namespace used to lookup and create plugins in the network.

setPluginNamespace

@NoException(value=true)
public void setPluginNamespace(@Cast(value="const char*")
                                                        BytePointer libNamespace)

setErrorRecorder

@NoException(value=true)
public void setErrorRecorder(IErrorRecorder recorder)

\brief 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()

getErrorRecorder

@NoException(value=true)
public IErrorRecorder getErrorRecorder()

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

Returns:

A pointer to the IErrorRecorder object that has been registered.

See Also:

setErrorRecorder()