box_embeddings.modules.pooling.intersection

Module Contents

hard_intersection_pooler(boxes: box_embeddings.parameterizations.box_tensor.BoxTensor, mask: torch.BoolTensor = None, dim: int = 0, keepdim: bool = False) → box_embeddings.parameterizations.box_tensor.BoxTensor

class HardIntersectionBoxPooler(dim: int = 0, keepdim: bool = False)

Bases: box_embeddings.modules.pooling.pooling.BoxPooler

Pools a box tensor using hard intersection operation

forward(self, box_tensor: box_embeddings.parameterizations.box_tensor.BoxTensor, mask: torch.BoolTensor = None) → box_embeddings.parameterizations.box_tensor.BoxTensor