tf.keras.utils.bounding_boxes.affine_transform

Applies an affine transformation to the bounding boxes.

tf.keras.utils.bounding_boxes.affine_transform(
    boxes,
    angle,
    translate_x,
    translate_y,
    scale,
    shear_x,
    shear_y,
    height,
    width,
    center_x=None,
    center_y=None,
    bounding_box_format='xyxy'
)

The height and width parameters are used to normalize the translation and scaling factors.

Args
boxes	The bounding boxes to transform, a tensor/array of shape (N, 4) or (batch_size, N, 4).
angle	Rotation angle in degrees.
translate_x	Horizontal translation fraction.
translate_y	Vertical translation fraction.
scale	Scaling factor.
shear_x	Shear angle in x-direction (degrees).
shear_y	Shear angle in y-direction (degrees).
height	Height of the image/data.
width	Width of the image/data.
center_x	x-coordinate of the transformation center (fraction).
center_y	y-coordinate of the transformation center (fraction).
bounding_box_format	The format of the input bounding boxes. Defaults to "xyxy".

Returns
The transformed bounding boxes, a tensor/array with the same shape as the input boxes.