[AutoAugment] Add Rotate operation.

Short Description Rotate is one of the operations used by AutoAugment. Currently it’s not present in keras_cv, so AutoAugment cannot be implemented.

What it does From Tensorlfow Addons docs: Rotate image(s) counterclockwise by the passed angle(s) in radians.

Papers AutoAugment: Learning Augmentation Strategies from Data arxiv link

Existing Implementations Tensorflow addons tensorflow/tpu

Other Information

• This is different from keras RandomRotation as we need to be able to control the probability at which the op is applied.
• Following standard Keras layer naming this could probably be called Rotation?

[sebastian-sz]

will you ever want to preprocess your inputs by ALWAYS rotating by X deg?

@LukeWood yes - this is actually the case in AutoAugment, where you want to rotate the image by X degrees based on subpolicy’s magnitude:

@quantumalaviya Thanks for linking #407 I didn’t know about it and it looks like exactly the thing needed.

I agree that this function doesn’t need to be a KLP and is good enough as a utility function. From my perspective this issue can be closed.

[LukeWood]

Cool, let’s leave it open until we can merge @quantumalaviya ’s PR and actually include the utility to rotate an image a fixed amount