Shortcuts

mmcv.ops.nms_rotated

mmcv.ops.nms_rotated(dets: torch.Tensor, scores: torch.Tensor, iou_threshold: float, labels: Optional[torch.Tensor] = None, clockwise: bool = True)Tuple[torch.Tensor, torch.Tensor][源代码]

Performs non-maximum suppression (NMS) on the rotated boxes according to their intersection-over-union (IoU).

Rotated NMS iteratively removes lower scoring rotated boxes which have an IoU greater than iou_threshold with another (higher scoring) rotated box.

参数
  • dets (torch.Tensor) – Rotated boxes in shape (N, 5). They are expected to be in (x_ctr, y_ctr, width, height, angle_radian) format.

  • scores (torch.Tensor) – scores in shape (N, ).

  • iou_threshold (float) – IoU thresh for NMS.

  • labels (torch.Tensor, optional) – boxes’ label in shape (N,).

  • clockwise (bool) – flag indicating whether the positive angular orientation is clockwise. default True. New in version 1.4.3.

返回

kept dets(boxes and scores) and indice, which is always the same data type as the input.

返回类型

tuple

Read the Docs v: stable
Versions
latest
stable
2.x
v2.0.1
v2.0.0
1.x
v1.7.1
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
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.