fix post_process.py (#3181)

PaddlePaddle · Jun 9, 2021 · 27d4a24 · 27d4a24
1 parent 31aa88c
commit 27d4a24
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppdet/modeling/post_process.py b/ppdet/modeling/post_process.py
@@ -78,7 +78,7 @@ def get_pred(self, bboxes, bbox_num, im_shape, scale_factor):
         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.