Skip to content

GitLab

Commit 58959e8d authored by Eric Jang's avatar Eric Jang Committed by TF Object Detection Team
Browse files

Add Arxiv link to AutoAugment paper. 

PiperOrigin-RevId: 374891049
parent 7da484db
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
research/object_detection/core/preprocessor.py
View file @ 58959e8d
...@@ -2888,10 +2888,12 @@ def random_patch_gaussian(image, ...@@ -2888,10 +2888,12 @@ def random_patch_gaussian(image,
return image return image
# TODO(barretzoph): Put in AutoAugment Paper link when paper is live.
def autoaugment_image(image, boxes, policy_name='v0'): def autoaugment_image(image, boxes, policy_name='v0'):
"""Apply an autoaugment policy to the image and boxes. """Apply an autoaugment policy to the image and boxes.
See "AutoAugment: Learning Augmentation Policies from Data" by Cubuk et al.,
2018, for further details. https://arxiv.org/abs/1805.09501
Args: Args:
image: rank 3 float32 tensor contains 1 image -> [height, width, channels] image: rank 3 float32 tensor contains 1 image -> [height, width, channels]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment