mmcls.datasets.pipelines.ColorTransform

class mmcls.datasets.pipelines.ColorTransform(magnitude, prob=0.5, random_negative_prob=0.5)

Adjust images color balance.

Parameters

• magnitude (int | float) – The magnitude used for color transform. A positive magnitude would enhance the color and a negative magnitude would make the image grayer. A magnitude=0 gives the origin img.

• prob (float) – The probability for performing ColorTransform therefore should be in range [0, 1]. Defaults to 0.5.

• random_negative_prob (float) – The probability that turns the magnitude negative, which should be in range [0,1]. Defaults to 0.5.