cerebras.modelzoo.data.vision.segmentation.transforms.color_transforms#

Classes

BrightnessMultiplicativeTransform

Augments the brightness of data.

ContrastAugmentationTransform

Augments the contrast of data :param contrast_range: (float, float): range from which to sample a random contrast that is applied to the data. If one value is smaller and one is larger than 1, half of the contrast modifiers will be >1 and the other half <1 (in the inverval that was specified) callable : must be contrast_range() -> float :param preserve_range: if True then the intensity values after contrast augmentation will be cropped to min and max values of the data before augmentation. :param per_channel: whether to use the same contrast modifier for all color channels or a separate one for each channel :param data_key: :param p_per_sample:.

GammaTransform

Augments by changing 'gamma' of the image (same as gamma correction in photos or computer monitors