Merges the different levels of the pyramid back to an image.
tfg.image.pyramid.merge(
levels, name=None
)
Args |
|---|
levels
|
A list containing tensors of shape [B, H_i, W_i, C], where B is
the batch size, H_i and W_i are the height and width of the image for the
level i, and C the number of channels of the image.
|
name
|
A name for this op that defaults to "pyramid_merge".
|
Returns |
|---|
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.
|
Raises |
|---|
ValueError
|
If the shape of the elements of levels is not supported.
|
View source on GitHub