Skip to content

GitLab

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "ab4990e4bfd79fe60815a3f872c5857df57798bb"
Unverified Commit c1748ef5 authored by Yezhen Cong's avatar Yezhen Cong Committed by GitHub
Browse files

[Fix] Add reference in`sample_valid_seeds` function used by ImVoteNet (#637) 

* Add reference in`sample_valid_seeds` function used by ImVoteNet

* add r to docstring
parent 8609b3b6
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
mmdet3d/models/detectors/imvotenet.py
View file @ c1748ef5
...@@ -10,7 +10,9 @@ from .base import Base3DDetector ...@@ -10,7 +10,9 @@ from .base import Base3DDetector
def sample_valid_seeds(mask, num_sampled_seed=1024): def sample_valid_seeds(mask, num_sampled_seed=1024):
"""Randomly sample seeds from all imvotes. r"""Randomly sample seeds from all imvotes.
Modified from `<https://github.com/facebookresearch/imvotenet/blob/a8856345146bacf29a57266a2f0b874406fd8823/models/imvotenet.py#L26>`_ # noqa
Args: Args:
mask (torch.Tensor): Bool tensor in shape ( mask (torch.Tensor): Bool tensor in shape (
......
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