Skip to main content

create_yuv

_static _TensorInfo.create_yuv(w : int, h : int, format : DataFmt, batch : int = 1)

Creates a TensorInfo object for a YUV image with the specified batch size, width, height, and format.

Parameters:

  • w (int) -- The width of each YUV image.

  • h (int) -- The height of each YUV image.

  • format (DataFmt) -- The format of the YUV image, defined in enum DataFmt.

  • batch (int _,_optional) -- The number of YUV images in the batch. Defaults to 1.

Returns:

tensor_info -- An object representing the information about a tensor that contains a batch of YUV images.

Return type:

TensorInfo

Raises:

RuntimeError -- If the YUV tensor creation fails.