vis4d.op.detect3d.bevformer.grid_mask

Grid mask for BEVFormer.

Classes

GridMask(use_h, use_w[, rotate, offset, ...])

Grid Mask Layer.

class GridMask(use_h, use_w, rotate=1, offset=False, ratio=0.5, mode=0, prob=1.0)[source]

Grid Mask Layer.

Init.

forward(x)[source]

Forward.

Return type:

Tensor