tfm.vision.augment.rotate_with_bboxes

Equivalent of PIL Rotate that rotates the image and bbox.

tfm.vision.augment.rotate_with_bboxes(
    image, bboxes, degrees, replace
)

image 3D uint8 Tensor.
bboxes 2D Tensor that is a list of the bboxes in the image. Each bbox has 4 elements (min_y, min_x, max_y, max_x) of type float.
degrees Float, a scalar angle in degrees to rotate all images by. If degrees is positive the image will be rotated clockwise otherwise it will be rotated counterclockwise.
replace A one or three value 1D tensor to fill empty pixels.

A tuple containing a 3D uint8 Tensor that will be the result of rotating image by degrees. The second element of the tuple is bboxes, where now the coordinates will be shifted to reflect the rotated image.

ValueError If applied to video.