Cutout

class mmpretrain.datasets.transforms.Cutout(shape=None, pad_val=128, prob=0.5, **kwargs)

擦除部分图像区域

参数:

• 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, see BaseAugTransform. 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.

transform(results)

Apply transform to results.