box_embeddings.modules.pooling.tf_bag_of_boxes

Module Contents

tf_bag_of_boxes_pooler(boxes: box_embeddings.parameterizations.tf_box_tensor.TFBoxTensor, mask: Optional[tensorflow.Tensor] = None, weights: Optional[tensorflow.Tensor] = None, dim: int = 0, keepdim: bool = False) → box_embeddings.parameterizations.tf_box_tensor.TFBoxTensor

class TFBagOfBoxesBoxPooler(dim: int = 0, keepdim: bool = False)

Bases: box_embeddings.modules.pooling.tf_pooling.TFBoxPooler

Pools a box tensor using hard intersection operation