embodichain.utils.warp.kernels.reshape_tiled_image

embodichain.utils.warp.kernels.reshape_tiled_image#

embodichain.utils.warp.kernels.reshape_tiled_image(tiled_image_buffer, batched_image, image_height, image_width, num_channels, num_tiles_x) = <warp.context.Kernel object>[source]#

Reshapes a tiled image into a batch of images.

This function reshapes the input tiled image buffer into a batch of images. The input image buffer is assumed to be tiled in the x and y directions. The output image is a batch of images with the specified height, width, and number of channels.

Parameters:
  • tiled_image_buffer (Any) – The input image buffer. Shape is (height * width * num_channels * num_cameras,).

  • batched_image (Any) – The output image. Shape is (num_cameras, height, width, num_channels).

  • image_width (int) – The width of the image.

  • image_height (int) – The height of the image.

  • num_channels (int) – The number of channels in the image.

  • num_tiles_x (int) – The number of tiles in x-direction.