diff --git a/.gitignore b/.gitignore index 36fea6908526258762cb6b41600b41c207d7433f..db7a3d642795a4372d502aef29002dcf180688f2 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 edd7cfad86610b87b68944ec16833245aebdcc7c..accef57eec70f3fabf39697e24117362193c16d1 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.