tfa.image.rotate

Rotate image(s) counterclockwise by the passed angle(s) in radians.

tfa.image.rotate(
    images: tfa.types.TensorLike,
    angles: tfa.types.TensorLike,
    interpolation: str = 'nearest',
    fill_mode: str = 'constant',
    name: Optional[str] = None,
    fill_value: tfa.types.TensorLike = 0.0
) -> tf.Tensor

Used in the notebooks

Used in the tutorials

images A tensor of shape (num_images, num_rows, num_columns, num_channels) (NHWC), (num_rows, num_columns, num_channels) (HWC), or (num_rows, num_columns) (HW).
angles A scalar angle to rotate all images by, or (if images has rank 4) a vector of length num_images, with an angle for each image in the batch.
interpolation Interpolation mode. Supported values: "nearest", "bilinear".
fill_mode Points outside the boundaries of the input are filled according to the given mode (one of {'constant', 'reflect', 'wrap', 'nearest'}).

  • reflect: (d c b a | a b c d | d c b a) The input is extended by reflecting about the edge of the last pixel.
  • constant: (k k k k | a b c d | k k k k) The input is extended by filling all values beyond the edge with the same constant value k = 0.
  • wrap: (a b c d | a b c d | a b c d) The input is extended by wrapping around to the opposite edge.
  • nearest: (a a a a | a b c d | d d d d) The input is extended by the nearest pixel.
fill_value a float represents the value to be filled outside the boundaries when fill_mode is "constant".
name The name of the op.

Image(s) with the same type and shape as images, rotated by the given angle(s). Empty space due to the rotation will be filled with zeros.

TypeError If images is an invalid type.