Equivalent of PIL Rotate that rotates the image and bbox.
tfm.vision.augment.rotate_with_bboxes(
image, bboxes, degrees, replace
)
Args |
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.
|
Returns |
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.
|
Raises |
ValueError
|
If applied to video.
|