From 27d4a24d31c9c68160acf77eeb1b33575aed7333 Mon Sep 17 00:00:00 2001 From: Jiang anqing Date: Wed, 9 Jun 2021 10:36:45 +0800 Subject: [PATCH] fix post_process.py (#3181) --- ppdet/modeling/post_process.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/modeling/post_process.py b/ppdet/modeling/post_process.py index 6ab9da6f4..42fbac7dc 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. -- GitLab