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