Skip to content

Commit 591a99b

Browse files
Correct min opset of CombinedNMS to be opset 12 (#1488)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent 9104dc8 commit 591a99b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tf2onnx/onnx_opset/tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1845,7 +1845,7 @@ def version_13(cls, ctx, node, **kwargs):
18451845
@tf_op(["CombinedNonMaxSuppression"])
18461846
class CombinedNonMaxSuppression:
18471847
@classmethod
1848-
def version_10(cls, ctx, node, **kwargs):
1848+
def version_12(cls, ctx, node, **kwargs):
18491849
# boxes.shape = [batch_size, num_boxes, (1 OR num_classes), 4]
18501850
# scores.shape = [batch_size, num_boxes, num_classes]
18511851
boxes, scores, max_per_class, max_total_size, iou_threshold, score_threshold = node.input

0 commit comments

Comments
 (0)