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.