Custom data augmentation with keras layers

Hi there. I was wondering how can I use keras’s layers to perform data augmentation.
For instance, I’d like to use: keras.layers.RandomBrightness() in the augment_image() function in the training script.

I tried to import them, but it looks like they do not exist. May I ask which Python and TensorFlow version is EI using in the backend?

Context/Use case:
Keras’s layers are more documented, more robust and easier to use w.r.t. using tf.image directly.

Thanks for your help.

Hi @berto

Are you trying to perform this in Expert mode or with the Python SDK?



Hi @Eoin,

Thank you for your reply.
I’m trying to do this in Expert mode.