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
andpost_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