tfm.vision.augment.ImageAugment

Image augmentation class for applying image distortions.

Methods

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(
 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`.