Cutout¶
- class mmpretrain.datasets.transforms.Cutout(shape=None, pad_val=128, prob=0.5, **kwargs)[source]¶
Cutout images.
- Parameters:
shape (int | tuple(int) | None) – Expected cutout shape (h, w). If given as a single value, the value will be used for both h and w. If None, generate from
magnitude_range
, seeBaseAugTransform
. Defaults to None.pad_val (int, Sequence[int]) – Pixel pad_val value for constant fill. If it is a sequence, it must have the same length with the image channels. Defaults to 128.
prob (float) – The probability for performing cutout therefore should be in range [0, 1]. Defaults to 0.5.
**kwargs – Other keyword arguments of
BaseAugTransform
.