tfm.vision.augment.ImageAugment

Image augmentation class for applying image distortions.

Methods

distort

View source

distort(
    image: tf.Tensor
) -> tf.Tensor

Given an image tensor, returns a distorted image with the same shape.

Expect the image tensor values are in the range [0, 255].

Args
image Tensor of shape [height, width, 3] or [num_frames, height, width, 3] representing an image or image sequence.

Returns
The augmented version of image.

distort_with_boxes

View source

distort_with_boxes(
    image: tf.Tensor, bboxes: tf.Tensor
) -> Tuple[tf.Tensor, tf.Tensor]

Distorts the image and bounding boxes.

Expect the image tensor values are in the range [0, 255].

Args
image Tensor of shape [height, width, 3] or [num_frames, height, width, 3] representing an image or image sequence.
bboxes Tensor of shape [num_boxes, 4] or [num_frames, num_boxes, 4] representing bounding boxes for an image or image sequence.

Returns
The augmented version of image and bboxes.