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 ¶