Shortcuts

mmcv.ops.points_in_polygons

mmcv.ops.points_in_polygons(points: torch.Tensor, polygons: torch.Tensor)torch.Tensor[source]

Judging whether points are inside polygons, which is used in the ATSS assignment for the rotated boxes.

It should be noted that when the point is just at the polygon boundary, the judgment will be inaccurate, but the effect on assignment is limited.

Parameters
  • points (torch.Tensor) – It has shape (B, 2), indicating (x, y). M means the number of predicted points.

  • polygons (torch.Tensor) – It has shape (M, 8), indicating (x1, y1, x2, y2, x3, y3, x4, y4). M means the number of ground truth polygons.

Returns

Return the result with the shape of (B, M), 1 indicates that the point is inside the polygon, 0 indicates that the point is outside the polygon.

Return type

torch.Tensor