fix bug (#756)

Co-authored-by: Jason <[email protected]>
PaddlePaddle · May 26, 2022 · 0881fec · 0881fec
1 parent ecb52a9
commit 0881fec
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/paddle2onnx/legacy/op_mapper/detection/multiclass_nms.py b/paddle2onnx/legacy/op_mapper/detection/multiclass_nms.py
@@ -74,6 +74,10 @@ def opset_10(cls, graph, node, **kw):
                     outputs=num_class,
                     axis=0)
             bbox_ids = []
+            if not isinstance(scores_list, list):
+                scores_list = [scores_list]
+            if not isinstance(bboxes_list, list):
+                bboxes_list = [bboxes_list]
             for i in range(num_class):
                 bbox_id = cls.nms(graph,
                                   node,