sources/gst-plugins/gst-nvdsvideotemplate/ds-sparse4d/infer/engine_builder.h File Reference

Go to the source code of this file.

Data Structures
struct	TRTDeleter
class	Logger

Macros
#define	TRT_MAJOR_VERSION   10
#define	TRT_MINOR_VERSION   3
#define	TRT_PATCH_VERSION   0

Typedefs
template<typename T >
using	TRTUniquePtr = std::unique_ptr< T, TRTDeleter >

Functions
bool	buildEngineFromOnnx (const std::string &onnxFilePath, const std::string &engineFilePath, int batchSize, int numSensors, bool fp16Mode, PrettyLogger *logger=nullptr)
	Builds a TensorRT engine from an ONNX model and saves it to the specified file path. More...

Variables
Logger	logger

Macro Definition Documentation

TRT_MAJOR_VERSION

#define TRT_MAJOR_VERSION   10

Definition at line 32 of file sources/gst-plugins/gst-nvdsvideotemplate/ds-sparse4d/infer/engine_builder.h.

TRT_MINOR_VERSION

#define TRT_MINOR_VERSION   3

Definition at line 33 of file sources/gst-plugins/gst-nvdsvideotemplate/ds-sparse4d/infer/engine_builder.h.

TRT_PATCH_VERSION

#define TRT_PATCH_VERSION   0

Definition at line 34 of file sources/gst-plugins/gst-nvdsvideotemplate/ds-sparse4d/infer/engine_builder.h.

Typedef Documentation

TRTUniquePtr

template<typename T >

using TRTUniquePtr = std::unique_ptr<T, TRTDeleter>

Definition at line 47 of file sources/gst-plugins/gst-nvdsvideotemplate/ds-sparse4d/infer/engine_builder.h.

Function Documentation

buildEngineFromOnnx()

bool buildEngineFromOnnx	(	const std::string &	onnxFilePath,
		const std::string &	engineFilePath,
		int	batchSize,
		int	numSensors,
		bool	fp16Mode,
		PrettyLogger *	logger = nullptr
	)

Builds a TensorRT engine from an ONNX model and saves it to the specified file path.

Compatible with TensorRT 10.3.

Parameters

onnxFilePath	Path to the input ONNX model
engineFilePath	Path where the output engine should be saved
batchSize	Sparse4D model batch size
numSensors	Number of sensors [Deepstream batch size]
fp16Mode	Whether to enable FP16 mode for faster inference
logger	Optional logger for structured logging

Returns

True if the engine was built successfully, false otherwise

Variable Documentation

logger

Logger logger