GitHub

tfm.vision.box_ops.gather_instances

Gather instances by indices.

tfm.vision.box_ops.gather_instances(
    selected_indices, instances, *aux_instances
)

Args
`selected_indices`	a Tensor of shape [batch, K] which indicates the selected indices in instance dimension (2nd dimension).
`instances`	a Tensor of shape [batch, N, ...] where the 2nd dimension is the instance dimension to be selected from.
`*aux_instances`	the additional Tensors whose shapes are in [batch, N, ...] which are the tensors to be selected from using the `selected_indices`.

Returns
`selected_instances`	the tensor of shape [batch, K, ...] which corresponds to the selected instances of the `instances` tensor.
`selected_aux_instances`	the additional tensors of shape [batch, K, ...] which corresponds to the selected instances of the `aus_instances` tensors.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2023-01-20 UTC.

Args