mmcv.ops.nms3d¶
- mmcv.ops.nms3d(boxes: Tensor, scores: Tensor, iou_threshold: float) 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: