From 76ab27803fce08f2a1b3743afff72966007e6d10 Mon Sep 17 00:00:00 2001
From: wangguanzhong <jerrywgz@126.com>
Date: Thu, 13 May 2021 14:05:30 +0800
Subject: [PATCH] fix return_index in nms (#2980)

---
 ppdet/modeling/ops.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/modeling/ops.py b/ppdet/modeling/ops.py
index dc47d1294..5b4986627 100644
--- a/ppdet/modeling/ops.py
+++ b/ppdet/modeling/ops.py
@@ -987,7 +987,7 @@ def multiclass_nms(bboxes,
                  'normalized', normalized)
         output, index, nms_rois_num = core.ops.multiclass_nms3(bboxes, scores,
                                                                rois_num, *attrs)
-        if return_index:
+        if not return_index:
             index = None
         return output, nms_rois_num, index
 
-- 
GitLab