Adding rgb feature name as an argument to the additional_augmentations fn.
Adding rgb feature name as an argument to the additional_augmentations fn.

PiperOrigin-RevId: 441508366

google-research/scenic