vis4d.data.transforms.random_erasing

Random erasing data augmentation.

Classes

RandomErasing(*[, in_keys, out_keys, ...])

Randomly erase a rectangular region in an image tensor.

class RandomErasing(*, in_keys=['images'], out_keys=['images'], sensors=None, same_on_batch=True, **kwargs)

Randomly erase a rectangular region in an image tensor.

__call__(images_list)

Execute the transform.

Return type:

list[Union[ndarray[Any, dtype[bool_]], ndarray[Any, dtype[float32]], ndarray[Any, dtype[float64]], ndarray[Any, dtype[int32]], ndarray[Any, dtype[int64]], ndarray[Any, dtype[uint8]], ndarray[Any, dtype[uint16]], ndarray[Any, dtype[uint32]]]]

do_erasing(images)

Execute the random erasing.

Return type:

Union[ndarray[Any, dtype[bool_]], ndarray[Any, dtype[float32]], ndarray[Any, dtype[float64]], ndarray[Any, dtype[int32]], ndarray[Any, dtype[int64]], ndarray[Any, dtype[uint8]], ndarray[Any, dtype[uint16]], ndarray[Any, dtype[uint32]]]