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.