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.

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".

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