ColorTransform¶
- class mmcls.datasets.transforms.ColorTransform(magnitude=None, prob=0.5, random_negative_prob=0.5, **kwargs)[源代码]¶
自动调整图像平衡
- 参数
magnitude (int | float | None) – 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. If None, generate from
magnitude_range
, seeBaseAugTransform
. Defaults to None.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.
**kwargs – Other keyword arguments of
BaseAugTransform
.