From 796874cfaaf63ee3d5ffcaca1da15937b7600c11 Mon Sep 17 00:00:00 2001 From: Guanghua Yu <742925032@qq.com> Date: Wed, 9 Jun 2021 10:54:09 +0800 Subject: [PATCH] cherry-pick some pr (#3333) --- .gitignore | 1 - ppdet/modeling/post_process.py | 3 +-- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/.gitignore b/.gitignore index 36fea6908..db7a3d642 100644 --- a/.gitignore +++ b/.gitignore @@ -61,7 +61,6 @@ coverage.xml # Sphinx documentation /docs/_build/ -*.json *.tar *.pyc diff --git a/ppdet/modeling/post_process.py b/ppdet/modeling/post_process.py index edd7cfad8..accef57ee 100644 --- a/ppdet/modeling/post_process.py +++ b/ppdet/modeling/post_process.py @@ -50,7 +50,6 @@ class BBoxPostProcess(nn.Layer): dtype='float32')) self.fake_bbox_num = paddle.to_tensor(np.array([1], dtype='int32')) - def forward(self, head_out, rois, im_shape, scale_factor): """ Decode the bbox and do NMS if needed. @@ -84,7 +83,7 @@ class BBoxPostProcess(nn.Layer): Currently only support bs = 1. 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. bbox_num (Tensor): The number of prediction boxes of each batch with shape [1], and is N. -- GitLab