 View source on GitHub |
Image augmentation class for applying image distortions.
Methods
distort
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
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. |