Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

Targer gather for dense object detector.



View source

Labels anchors with ground truth inputs.

B: batch_size N: number of groundtruth boxes.

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.

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

ValueError If labels is higher than rank 3.