box_embeddings.modules.pooling.pooling
¶
Module Contents¶
- class BoxPooler¶
Bases:
torch.nn.Module
,box_embeddings.common.registrable.Registrable
Base class for Box Pooling
- abstract forward(self, box_tensor: box_embeddings.parameterizations.BoxTensor, **kwargs: Any) box_embeddings.parameterizations.BoxTensor ¶