box_embeddings.modules.volume.tf_volume

Module Contents

eps

class TFVolume(log_scale: bool = True, **kwargs: Any)

Bases: tensorflow.Module, box_embeddings.common.registrable.Registrable

Base volume class

default_implementation = hard

abstract forward(self, box_tensor: box_embeddings.parameterizations.tf_box_tensor.TFBoxTensor) → tensorflow.Tensor

Base implementation is hard (ReLU) volume.

Parameters

box_tensor – Input box tensor

Raises

NotImplementedError – base class

tf_hard_volume(box_tensor: box_embeddings.parameterizations.tf_box_tensor.TFBoxTensor) → tensorflow.Tensor

Volume of boxes. Returns 0 where boxes are flipped.

Parameters

box_tensor – input

Returns

Tensor of shape (…, ) when self has shape (…, 2, num_dims)

tf_log_hard_volume(box_tensor: box_embeddings.parameterizations.tf_box_tensor.TFBoxTensor) → tensorflow.Tensor

class TFHardVolume(log_scale: bool = True, **kwargs: Any)

Bases: TFVolume

Hard ReLU based volume.