Transform Image Data Depending on Label

Is there a way to augment my data depending on their labels, ie. lets say I have trees, water, ice in my dataset and I would want to apply different contrast for trees (eg scale=[0, 1]) and apply some other contrast on water and ice as the result very much depends on their colors(eg scale=[0.5, 1])?

Thanks a lot