box_embeddings.modules.volume.hard_volume

Module Contents

eps
hard_volume(box_tensor: box_embeddings.parameterizations.BoxTensor, log_scale: bool = True) torch.Tensor

Volume of boxes. Returns 0 where boxes are flipped.

Parameters
  • box_tensor – input

  • log_scale – Whether the output should be in log scale or not.

Returns

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

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

Bases: box_embeddings.modules.volume._volume._Volume

Hard ReLU based volume.

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

Hard ReLU base volume.

Parameters

box_tensor – TODO

Returns

torch.Tensor