tfm.vision.target_gather.TargetGather

Targer gather for dense object detector.

Methods

__call__

View source

__call__(
    labels, match_indices, mask=None, mask_val=0.0
)

Labels anchors with ground truth inputs.

B: batch_size N: number of groundtruth boxes.

Args
labels An integer tensor with shape [N, dims] or [B, N, ...] representing groundtruth labels.
match_indices An integer tensor with shape [M] or [B, M] representing match label index.
mask An boolean tensor with shape [M, dims] or [B, M,...] representing match labels.
mask_val An integer to fill in for mask.

Returns
target An integer Tensor with shape [M] or [B, M]

Raises
ValueError If labels is higher than rank 3.