Skip to content

Commit 8e41270

Browse files
authored
Merge pull request #895 from jignparm/jignparm/fix_nonmaxsuppression
Fix NonMaxSuppression
2 parents 212e119 + 539a175 commit 8e41270

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tf2onnx/onnx_opset/tensor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1488,10 +1488,10 @@ def version_10(cls, ctx, node, **kwargs):
14881488
pad_val = ctx.make_node("Cast", inputs=[relu_op.output[0]], attr={"to": onnx_pb.TensorProto.INT64})
14891489
pad_op = ctx.make_node("Pad", inputs=[squeeze_op.output[0], pad_val.output[0]])
14901490
ctx.make_node("Cast", inputs=pad_op.output, name="cast_A", attr={"to": onnx_pb.TensorProto.INT32},
1491-
outputs=[node.output[0]], dtypes=dtypes[0], shapes=shapes[0])
1491+
outputs=[node.output[0]], dtypes=dtypes[0], shapes=shapes[0], op_name_scope=node.name)
14921492
reduce_op = ctx.make_node("ReduceSum", inputs=shape_op.output, attr={"axes": [0], "keepdims": 0})
14931493
ctx.make_node("Cast", inputs=[reduce_op.output[0]], name="cast_B", attr={"to": onnx_pb.TensorProto.INT32},
1494-
outputs=[node.output[1]], dtypes=dtypes[1], shapes=shapes[1])
1494+
outputs=[node.output[1]], dtypes=dtypes[1], shapes=shapes[1], op_name_scope=node.name)
14951495
else:
14961496
ctx.make_node("Cast", inputs=squeeze_op.output, attr={"to": onnx_pb.TensorProto.INT32},
14971497
name=node.name, outputs=node.output, dtypes=dtypes[0], shapes=shapes[0])

0 commit comments

Comments
 (0)