Save the date! Google I/O returns May 18-20 Register now

tfa.image.cutout

Apply cutout to images.

This operation applies a (mask_height x mask_width) mask of zeros to a location within images specified by the offset. The pixel values filled in will be of the value constant_values. The located where the mask will be applied is randomly chosen uniformly over the whole images.

images A tensor of shape (batch_size, height, width, channels) (NHWC).
mask_size Specifies how big the zero mask that will be generated is that is applied to the images. The mask will be of size (mask_height x mask_width). Note: mask_size should be divisible by 2.
offset A tuple of (height, width) or (batch_size, 2)
constant_values What pixel value to fill in the images in the area that has the cutout mask applied to it.

A Tensor of the same shape and dtype as images.

InvalidArgumentError if mask_size can't be divisible by 2.