Shortcuts

mmcv.ops.nms3d_normal

mmcv.ops.nms3d_normal(boxes: torch.Tensor, scores: torch.Tensor, iou_threshold: float)torch.Tensor[source]

Normal 3D NMS function GPU implementation. 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, 7). ([x, y, z, dx, dy, dz, heading]).

  • scores (torch.Tensor) – Scores of predicted boxes with shape (N).

  • iou_threshold (float) – Overlap threshold of NMS.

Returns

Remaining indices with scores in descending order.

Return type

torch.Tensor

Read the Docs v: 2.x
Versions
master
latest
2.x
1.x
v1.7.0
v1.6.2
v1.6.1
v1.6.0
v1.5.3
v1.5.2_a
v1.5.1
v1.5.0
v1.4.8
v1.4.7
v1.4.6
v1.4.5
v1.4.4
v1.4.3
v1.4.2
v1.4.1
v1.4.0
v1.3.18
v1.3.17
v1.3.16
v1.3.15
v1.3.14
v1.3.13
v1.3.12
v1.3.11
v1.3.10
v1.3.9
v1.3.8
v1.3.7
v1.3.6
v1.3.5
v1.3.4
v1.3.3
v1.3.2
v1.3.1
v1.3.0
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.