update output ports count depending on opset version (#4439)

This commit is contained in:
Yegor Kruglov authored and GitHub committed 2021-02-26 09:18:03 +03:00
1 parent 53a96f4a5a
commit f7b6ad710f
1 file changed
+1 -1
+1 -1
View File
@@ -31,7 +31,7 @@ class ProposalOp(Op):
'post_nms_topn': 300, # default in caffe-shared
'infer': ProposalOp.proposal_infer,
'in_ports_count': 3,
'out_ports_count': 2,
'out_ports_count': 1 if attrs.get('version') == 'opset1' else 2,
'normalize': False,
'clip_before_nms': True,
'clip_after_nms': False,