NVIDIA DeepStream SDK API Reference

9.2 Release
TRTConfigParams Class Reference

Detailed Description

Stores TensorRT model and engine configuration.

Definition at line 32 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

Inheritance diagram for TRTConfigParams:
Collaboration diagram for TRTConfigParams:

Public Member Functions

 TRTConfigParams ()
 Construct a TRTConfigParams instance. More...
 
 TRTConfigParams (const std::string &configFilePath, const std::string moduleName, const std::string submoduleName)
 Construct a TRTConfigParams instance. More...
 
 TRTConfigParams (const YAML::Node &fn)
 Construct a TRTConfigParams instance. More...
 
void read (const YAML::Node &fn) override
 Load configuration values from a YAML node. More...
 
 TRTConfigParams ()
 Construct a TRTConfigParams instance. More...
 
 TRTConfigParams (const std::string &configFilePath, const std::string moduleName, const std::string submoduleName)
 Construct a TRTConfigParams instance. More...
 
 TRTConfigParams (const YAML::Node &fn)
 Construct a TRTConfigParams instance. More...
 
void read (const YAML::Node &fn) override
 Load configuration values from a YAML node. More...
 

Static Public Member Functions

static std::string resolvePathRelativeToConfig (const std::string &configFilePath, const std::string &filePath)
 Resolve a configured path relative to the tracker configuration file. More...
 
static std::string resolvePathRelativeToConfig (const std::string &configFilePath, const std::string &filePath)
 Resolve a configured path relative to the tracker configuration file. More...
 
static const YAML::Node getConfigRoot (const std::string &configFilePath, const bool printMessage=false)
 Load the root YAML node from a tracker configuration file. More...
 
static const YAML::Node getConfigRoot (const std::string &configFilePath, const bool printMessage=false)
 Load the root YAML node from a tracker configuration file. More...
 

Data Fields

int m_maxBatchSize
 Maximum batch size for reid network. More...
 
int m_inputOrder
 Input dimension order: 0: NCHW, 1: NHWC. More...
 
int m_width
 Input image patch width. More...
 
int m_height
 Input image patch height. More...
 
int m_channel
 Input image patch channel. More...
 
int m_gpuId
 Gpu id. More...
 
std::vector< int > m_inferDims
 Reid input dimension. More...
 
size_t m_workspaceSize
 Work space for reid network in MiB. More...
 
std::string m_tltEncodedModelFilePath
 Reid network tao model path. More...
 
std::string m_tltModelKey
 Tao model key. More...
 
std::string m_onnxFilePath
 Reid network onnx model path. More...
 
std::string m_engineFilePath
 Reid engine path. More...
 
std::string m_calibrationTableFilePath
 Reid int8 calibration table file path. More...
 
nvinfer1::DataType m_networkMode
 Reid network precision. More...
 
int m_builderOptimizationLevel
 Engine builder optimization level. More...
 
int m_debugVerboseLevel
 Debug message verbose mode level. More...
 

Constructor & Destructor Documentation

◆ TRTConfigParams() [1/6]

TRTConfigParams::TRTConfigParams ( )

Construct a TRTConfigParams instance.

◆ TRTConfigParams() [2/6]

TRTConfigParams::TRTConfigParams ( const std::string &  configFilePath,
const std::string  moduleName,
const std::string  submoduleName 
)

Construct a TRTConfigParams instance.

Parameters
[in]configFilePathPath to the tracker configuration file.
[in]moduleNameConfiguration module name.
[in]submoduleNameConfiguration submodule name.

◆ TRTConfigParams() [3/6]

TRTConfigParams::TRTConfigParams ( const YAML::Node &  fn)

Construct a TRTConfigParams instance.

Parameters
[in]fnParsed YAML node.

◆ TRTConfigParams() [4/6]

TRTConfigParams::TRTConfigParams ( )

Construct a TRTConfigParams instance.

◆ TRTConfigParams() [5/6]

TRTConfigParams::TRTConfigParams ( const std::string &  configFilePath,
const std::string  moduleName,
const std::string  submoduleName 
)

Construct a TRTConfigParams instance.

Parameters
[in]configFilePathPath to the tracker configuration file.
[in]moduleNameConfiguration module name.
[in]submoduleNameConfiguration submodule name.

◆ TRTConfigParams() [6/6]

TRTConfigParams::TRTConfigParams ( const YAML::Node &  fn)

Construct a TRTConfigParams instance.

Parameters
[in]fnParsed YAML node.

Member Function Documentation

◆ getConfigRoot() [1/2]

static const YAML::Node NvTrackerParams::getConfigRoot ( const std::string &  configFilePath,
const bool  printMessage = false 
)
inlinestaticinherited

◆ getConfigRoot() [2/2]

static const YAML::Node NvTrackerParams::getConfigRoot ( const std::string &  configFilePath,
const bool  printMessage = false 
)
inlinestaticinherited

Load the root YAML node from a tracker configuration file.

Parameters
[in]configFilePathPath to the tracker configuration file.
[in]printMessageEnables warning output when the configuration cannot be loaded.
Returns
Loaded YAML root node, or an empty node when loading fails.

Definition at line 121 of file 9.2/sources/libs/nvmultiobjecttracker/include/modules/NvMultiObjectTracker/NvTrackerParams.hpp.

References CLOG.

◆ read() [1/2]

void TRTConfigParams::read ( const YAML::Node &  fn)
overridevirtual

Load configuration values from a YAML node.

Parameters
[in]fnParsed YAML node.

Implements NvTrackerParams.

◆ read() [2/2]

void TRTConfigParams::read ( const YAML::Node &  fn)
overridevirtual

Load configuration values from a YAML node.

Parameters
[in]fnParsed YAML node.

Implements NvTrackerParams.

◆ resolvePathRelativeToConfig() [1/2]

static std::string NvTrackerParams::resolvePathRelativeToConfig ( const std::string &  configFilePath,
const std::string &  filePath 
)
inlinestaticinherited

Resolve a configured path relative to the tracker configuration file.

Existing paths are canonicalized, while nonexistent relative paths remain joined to the configuration directory.

Parameters
[in]configFilePathPath to the tracker configuration file.
[in]filePathAbsolute or configuration-relative path to resolve.
Returns
Resolved path, or the original path when the configuration path cannot be resolved.

Definition at line 55 of file sources/libs/nvmultiobjecttracker/include/modules/NvMultiObjectTracker/NvTrackerParams.hpp.

Referenced by CommunicatorParams::CommunicatorParams(), and MqttCommunicatorParams::MqttCommunicatorParams().

◆ resolvePathRelativeToConfig() [2/2]

static std::string NvTrackerParams::resolvePathRelativeToConfig ( const std::string &  configFilePath,
const std::string &  filePath 
)
inlinestaticinherited

Resolve a configured path relative to the tracker configuration file.

Existing paths are canonicalized, while nonexistent relative paths remain joined to the configuration directory.

Parameters
[in]configFilePathPath to the tracker configuration file.
[in]filePathAbsolute or configuration-relative path to resolve.
Returns
Resolved path, or the original path when the configuration path cannot be resolved.

Definition at line 55 of file 9.2/sources/libs/nvmultiobjecttracker/include/modules/NvMultiObjectTracker/NvTrackerParams.hpp.

Field Documentation

◆ m_builderOptimizationLevel

int TRTConfigParams::m_builderOptimizationLevel

Engine builder optimization level.

Definition at line 73 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_calibrationTableFilePath

std::string TRTConfigParams::m_calibrationTableFilePath

Reid int8 calibration table file path.

Definition at line 70 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_channel

int TRTConfigParams::m_channel

Input image patch channel.

Definition at line 62 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_debugVerboseLevel

int TRTConfigParams::m_debugVerboseLevel

Debug message verbose mode level.

Definition at line 74 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_engineFilePath

std::string TRTConfigParams::m_engineFilePath

◆ m_gpuId

int TRTConfigParams::m_gpuId

◆ m_height

int TRTConfigParams::m_height

Input image patch height.

Definition at line 61 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_inferDims

std::vector< int > TRTConfigParams::m_inferDims

◆ m_inputOrder

int TRTConfigParams::m_inputOrder

Input dimension order: 0: NCHW, 1: NHWC.

Definition at line 59 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_maxBatchSize

int TRTConfigParams::m_maxBatchSize

Maximum batch size for reid network.

Definition at line 58 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_networkMode

nvinfer1::DataType TRTConfigParams::m_networkMode

Reid network precision.

Integer 0: FP32 1: INT8 2: FP16.

Definition at line 71 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_onnxFilePath

std::string TRTConfigParams::m_onnxFilePath

Reid network onnx model path.

Definition at line 68 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_tltEncodedModelFilePath

std::string TRTConfigParams::m_tltEncodedModelFilePath

Reid network tao model path.

Definition at line 66 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_tltModelKey

std::string TRTConfigParams::m_tltModelKey

◆ m_width

int TRTConfigParams::m_width

Input image patch width.

Definition at line 60 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.

◆ m_workspaceSize

size_t TRTConfigParams::m_workspaceSize

Work space for reid network in MiB.

Definition at line 65 of file sources/libs/nvmultiobjecttracker/include/modules/FrameTransform/trtModule.hpp.


The documentation for this class was generated from the following file: