diff --git a/ppdet/modeling/ops.py b/ppdet/modeling/ops.py index dc47d12945739cfd45615b2dc830d5a13482c891..5b49866277e8ae5900d499ad87f2d1bedb63c4c7 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