batched_nms {torchvision} | R Documentation |
Batched Non-maximum Suppression (NMS)
Description
Performs non-maximum suppression in a batched fashion. Each index value correspond to a category, and NMS will not be applied between elements of different categories.
Usage
batched_nms(boxes, scores, idxs, iou_threshold)
Arguments
boxes |
(Tensor[N, 4]): boxes where NMS will be performed. They are expected to be
in
|
scores |
(Tensor[N]): scores for each one of the boxes |
idxs |
(Tensor[N]): indices of the categories for each one of the boxes. |
iou_threshold |
(float): discards all overlapping boxes with IoU > |
Value
keep (Tensor): int64 tensor with the indices of the elements that have been kept by NMS, sorted in decreasing order of scores
[Package torchvision version 0.7.0 Index]