Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "255e993b5db0d4dd4862f74de519ef50830b267f"
Unverified Commit 0ebd9757 authored by filaPro's avatar filaPro Committed by GitHub
Browse files

[Fix] box3d_multiclass_nms docstring (#398)

parent 7c300725
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmdet3d/core/post_processing/box3d_nms.py
View file @ 0ebd9757
...@@ -18,9 +18,9 @@ def box3d_multiclass_nms(mlvl_bboxes, ...@@ -18,9 +18,9 @@ def box3d_multiclass_nms(mlvl_bboxes,
mlvl_bboxes (torch.Tensor): Multi-level boxes with shape (N, M). mlvl_bboxes (torch.Tensor): Multi-level boxes with shape (N, M).
M is the dimensions of boxes. M is the dimensions of boxes.
mlvl_bboxes_for_nms (torch.Tensor): Multi-level boxes with shape mlvl_bboxes_for_nms (torch.Tensor): Multi-level boxes with shape
(N, 4). N is the number of boxes. (N, 5) ([x1, y1, x2, y2, ry]). N is the number of boxes.
mlvl_scores (torch.Tensor): Multi-level boxes with shape mlvl_scores (torch.Tensor): Multi-level boxes with shape
(N, ). N is the number of boxes. (N, C + 1). N is the number of boxes. C is the number of classes.
score_thr (float): Score thredhold to filter boxes with low score_thr (float): Score thredhold to filter boxes with low
confidence. confidence.
max_num (int): Maximum number of boxes will be kept. max_num (int): Maximum number of boxes will be kept.
......
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