mmcv.ops.nms_normal_bev¶
- mmcv.ops.nms_normal_bev(boxes: torch.Tensor, scores: torch.Tensor, thresh: float) → torch.Tensor[source]¶
Normal NMS function GPU implementation (for BEV boxes).
The overlap of two boxes for IoU calculation is defined as the exact overlapping area of the two boxes WITH their yaw angle set to 0.
- Parameters
boxes (torch.Tensor) – Input boxes with shape (N, 5) ([x1, y1, x2, y2, ry]).
scores (torch.Tensor) – Scores of predicted boxes with shape (N,).
thresh (float) – Overlap threshold of NMS.
- Returns
Remaining indices with scores in descending order.
- Return type