UFF Parser¶
-
tensorrt.
UffInputOrder
¶ The different possible supported input orders.
Members:
NCHW
NHWC
NC
-
class
tensorrt.
UffParser
(self: tensorrt.tensorrt.UffParser) → None¶ This class is used for parsing models described using the UFF format.
- Variables
uff_required_version_major –
int
Version Major of the UFF.uff_required_version_minor –
int
Version Minor of the UFF.uff_required_version_patch –
int
Version Patch of the UFF.plugin_factory –
IUffPluginFactory
used to create the user defined plugins.plugin_factory_ext –
IUffPluginFactoryExt
used to create the user defined pluginExts.plugin_namespace –
str
The namespace used to lookup and create plugins in the network.
-
__del__
(self: tensorrt.tensorrt.UffParser) → None¶
-
__exit__
(exc_type, exc_value, traceback)¶ Destroy this object, freeing all memory associated with it. This should be called to ensure that the object is cleaned up properly. Equivalent to invoking
__del__()
-
__init__
(self: tensorrt.tensorrt.UffParser) → None¶
-
parse
(self: tensorrt.tensorrt.UffParser, file: str, network: tensorrt.tensorrt.INetworkDefinition, weights_type: tensorrt.tensorrt.DataType = DataType.FLOAT) → bool¶ Parse a UFF file.
- Parameters
file – File name of the UFF file.
network – Network in which the
UffParser
will fill the layers.weights_type – The type on which the weights will be transformed in.
- Returns
True if the UFF file is parsed without error.
-
parse_buffer
(self: tensorrt.tensorrt.UffParser, buffer: buffer, network: tensorrt.tensorrt.INetworkDefinition, weights_type: tensorrt.tensorrt.DataType = DataType.FLOAT) → bool¶ Parse a UFF buffer - useful if the file is already live in memory.
- Parameters
buffer – The UFF buffer.
network – Network in which the UFFParser will fill the layers.
weights_type – The type on which the weights will be transformed in.
- Returns
True if the UFF buffer is parsed without error.
-
register_input
(self: tensorrt.tensorrt.UffParser, name: str, shape: tensorrt.tensorrt.Dims, order: tensorrt.tensorrt.UffInputOrder = UffInputOrder.NCHW) → bool¶ Register an input name of a UFF network with the associated Dimensions.
- Parameters
name – Input name.
shape – Input shape.
order – Input order on which the framework input was originally.
- Returns
True if the name registers without error.
-
register_output
(self: tensorrt.tensorrt.UffParser, name: str) → bool¶ Register an output name of a UFF network.
- Parameters
output_name – Output name.
- Returns
True if the name registers without error.