box_embeddings.modules.tf_embedding

Module Contents

class TFBoxEmbedding(num_embeddings: int, embedding_dim: int, box_factory: box_embeddings.parameterizations.tf_box_tensor.TFBoxFactory = None, box_initializer: box_embeddings.initializations.tf_initializer.TFBoxInitializer = None, **kwargs: Any)

Bases: tensorflow.keras.layers.Embedding

Embedding which returns boxes instead of vectors

property all_boxes(self) → box_embeddings.parameterizations.tf_box_tensor.TFBoxTensor

get_bounding_box(self) → box_embeddings.parameterizations.tf_box_tensor.TFBoxTensor