box_embeddings.modules.embedding

Module Contents

class BoxEmbedding(num_embeddings: int, embedding_dim: int, box_factory: box_embeddings.parameterizations.box_tensor.BoxFactory = None, box_initializer: box_embeddings.initializations.initializer.BoxInitializer = None, **kwargs: Any)

Bases: torch.nn.Embedding

Embedding which returns boxes instead of vectors

forward(self, inputs: torch.Tensor) → box_embeddings.parameterizations.box_tensor.BoxTensor

property all_boxes(self) → box_embeddings.parameterizations.box_tensor.BoxTensor

get_bounding_box(self) → box_embeddings.parameterizations.box_tensor.BoxTensor