box_embeddings.initializations.uniform_boxes

Module Contents

uniform_boxes(dimensions: int, num_boxes: int, minimum: float = 0.0, maximum: float = 1.0, delta_min: float = 0.01, delta_max: float = 0.5) → Tuple[torch.Tensor, torch.Tensor]

Creates uniform boxes such that each box is inside the bounding box defined by (minimum,maximum) in each dimension.

Parameters

• dimensions – number of dimensions for the box

• num_boxes – number of boxes to be created

• minimum – min dimension of the bounding box in each dimension.

• maximum – maximum dimension of the bounding box in each dimension.

• delta_min – TODO

• delta_max – TODO

Returns

z, Z tensors

Raises

ValueError – TODO

class UniformBoxInitializer(dimensions: int, num_boxes: int, box_type_factory: box_embeddings.parameterizations.box_tensor.BoxFactory, minimum: float = 0.0, maximum: float = 1.0, delta_min: float = 0.01, delta_max: float = 0.5)

Bases: box_embeddings.initializations.initializer.BoxInitializer

Docstring for UniformBoxInitializer.