TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfg.image.pyramid.split

Generates the different levels of the pyramid.

tfg.image.pyramid.split(
    image,
    num_levels,
    name=None
)

Defined in image/pyramid.py.

Args:

  • image: A tensor of shape [B, H, W, C], where B is the batch size, H the height of the image, W the width of the image, and C the number of channels of the image.
  • num_levels: The number of levels to generate.
  • name: A name for this op that defaults to "pyramid_split".

Returns:

A list containing num_levels tensors of shape [B, H_i, W_i, C], where H_i and W_i are the height and width of the image for the level i.

Raises:

  • ValueError: If the shape of image is not supported.