mmcv.ops.convex_iou

mmcv.ops.convex_iou(pointsets: torch.Tensor, polygons: torch.Tensor) → torch.Tensor

Return intersection-over-union (Jaccard index) between point sets and polygons.

参数

• pointsets (torch.Tensor) – It has shape (N, 18), indicating (x1, y1, x2, y2, …, x9, y9) for each row.

• polygons (torch.Tensor) – It has shape (K, 8), indicating (x1, y1, x2, y2, x3, y3, x4, y4) for each row.

返回

Return the ious between point sets and polygons with the shape (N, K).

返回类型

torch.Tensor