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