mmcv.ops.nms3d¶
- mmcv.ops.nms3d(boxes: torch.Tensor, scores: torch.Tensor, iou_threshold: float) → torch.Tensor[source]¶
3D NMS function GPU implementation (for BEV boxes).
- Parameters
boxes (torch.Tensor) – Input boxes with the shape of (N, 7) ([x, y, z, dx, dy, dz, heading]).
scores (torch.Tensor) – Scores of boxes with the shape of (N).
iou_threshold (float) – Overlap threshold of NMS.
- Returns
Indexes after NMS.
- Return type