Skip to content

GitLab

Unverified Commit 50d15052 authored by gy77's avatar gy77 Committed by GitHub
Browse files

Add type hint in mmcv/ops/points_in_polygons.py (#2013)

parent 335199db
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
mmcv/ops/points_in_polygons.py
View file @ 50d15052
import torch import torch
from torch import Tensor
from ..utils import ext_loader from ..utils import ext_loader
ext_module = ext_loader.load_ext('_ext', ['points_in_polygons_forward']) ext_module = ext_loader.load_ext('_ext', ['points_in_polygons_forward'])
def points_in_polygons(points, polygons): def points_in_polygons(points: Tensor, polygons: Tensor) -> Tensor:
"""Judging whether points are inside polygons, which is used in the ATSS """Judging whether points are inside polygons, which is used in the ATSS
assignment for the rotated boxes. assignment for the rotated boxes.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment