未验证 提交 796874cf 编写于 作者: G Guanghua Yu 提交者: GitHub

cherry-pick some pr (#3333)

上级 e5ebac11
...@@ -61,7 +61,6 @@ coverage.xml ...@@ -61,7 +61,6 @@ coverage.xml
# Sphinx documentation # Sphinx documentation
/docs/_build/ /docs/_build/
*.json
*.tar *.tar
*.pyc *.pyc
......
...@@ -50,7 +50,6 @@ class BBoxPostProcess(nn.Layer): ...@@ -50,7 +50,6 @@ class BBoxPostProcess(nn.Layer):
dtype='float32')) dtype='float32'))
self.fake_bbox_num = paddle.to_tensor(np.array([1], dtype='int32')) self.fake_bbox_num = paddle.to_tensor(np.array([1], dtype='int32'))
def forward(self, head_out, rois, im_shape, scale_factor): def forward(self, head_out, rois, im_shape, scale_factor):
""" """
Decode the bbox and do NMS if needed. Decode the bbox and do NMS if needed.
...@@ -84,7 +83,7 @@ class BBoxPostProcess(nn.Layer): ...@@ -84,7 +83,7 @@ class BBoxPostProcess(nn.Layer):
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.
先完成此消息的编辑!
想要评论请 注册