PyTorch Forums
Torchvision.ops.nms(boxes: torch.Tensor, scores: torch.Tensor, iou_threshold: float)
vision
raoshenglong
(shr)
January 19, 2021, 9:38am
1
I wonder this function only return the selected indices,how can we get the exactly selected num?