diff --git a/ppdet/modeling/post_process.py b/ppdet/modeling/post_process.py index 6ab9da6f4d67df01b7a4541130e78ba0fa6f9dee..42fbac7dcfbd3ecb726d29765009f880cbefefb2 100644 --- a/ppdet/modeling/post_process.py +++ b/ppdet/modeling/post_process.py @@ -78,7 +78,7 @@ class BBoxPostProcess(object): 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.