trtutils.impls.onnx_models module

ONNX model implementations.

:attribute:`YOLO_PREPROC_BASE`

The path to the ONNX model for YOLO preprocessing.

Type:

Path

trtutils.impls.onnx_models.build_yolo_preproc(input_shape: tuple[int, int], output_dtype: dtype) → Path[source]

Build a YOLO preproc TensorRT engine.

Parameters:

  • input_shape (tuple[int, int]) – The (width, height) of the YOLO network.

  • output_dtype (np.dtype) – The datatype to return, which the YOLO network will take as input.

Returns:

The path to the compiled engine.

Return type:

Path