Shortcuts

mmcv.ops.nms_bev

mmcv.ops.nms_bev(boxes: torch.Tensor, scores: torch.Tensor, thresh: float, pre_max_size: Optional[int] = None, post_max_size: Optional[int] = None)torch.Tensor[source]

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. In this function, one can also set pre_max_size and post_max_size.

Parameters
  • boxes (torch.Tensor) – Input boxes with the shape of (N, 5) ([x1, y1, x2, y2, ry]).

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

  • thresh (float) – Overlap threshold of NMS.

  • pre_max_size (int, optional) – Max size of boxes before NMS. Default: None.

  • post_max_size (int, optional) – Max size of boxes after NMS. Default: None.

Returns

Indexes after NMS.

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.