未验证 提交 27d4a24d 编写于 作者: J Jiang anqing 提交者: GitHub

fix post_process.py (#3181)

上级 31aa88c8
...@@ -78,7 +78,7 @@ class BBoxPostProcess(object): ...@@ -78,7 +78,7 @@ class BBoxPostProcess(object):
Currently only support bs = 1. Currently only support bs = 1.
Args: Args:
bbox_pred (Tensor): The output bboxes with shape [N, 6] after decode bboxes (Tensor): The output bboxes with shape [N, 6] after decode
and NMS, including labels, scores and bboxes. and NMS, including labels, scores and bboxes.
bbox_num (Tensor): The number of prediction boxes of each batch with bbox_num (Tensor): The number of prediction boxes of each batch with
shape [1], and is N. shape [1], and is N.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册