Shortcuts

mmcv.ops.nms_match

mmcv.ops.nms_match(dets: Union[torch.Tensor, numpy.ndarray], iou_threshold: float)List[Union[numpy.ndarray, torch.Tensor]][源代码]

Matched dets into different groups by NMS.

NMS match is Similar to NMS but when a bbox is suppressed, nms match will record the indice of suppressed bbox and form a group with the indice of kept bbox. In each group, indice is sorted as score order.

参数
  • dets (torch.Tensor | np.ndarray) – Det boxes with scores, shape (N, 5).

  • iou_threshold (float) – IoU thresh for NMS.

返回

The outer list corresponds different matched group, the inner Tensor corresponds the indices for a group in score order.

返回类型

list[torch.Tensor | np.ndarray]

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.