From 1066d4551f1f6e98c1b59a1905181f4f3edd9d07 Mon Sep 17 00:00:00 2001 From: Chenhu Wang Date: Fri, 10 Jun 2022 10:18:28 +0800 Subject: [PATCH] fix_nms_ops_transformation (#11794) * fix_nms_ops_transformation * replace node when 5-9 --- .../op_conversions/convert_previous_nms_to_nms_9.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/common/transformations/src/transformations/op_conversions/convert_previous_nms_to_nms_9.cpp b/src/common/transformations/src/transformations/op_conversions/convert_previous_nms_to_nms_9.cpp index 0a4c49f1bd7..f1cb0cc3302 100644 --- a/src/common/transformations/src/transformations/op_conversions/convert_previous_nms_to_nms_9.cpp +++ b/src/common/transformations/src/transformations/op_conversions/convert_previous_nms_to_nms_9.cpp @@ -186,7 +186,11 @@ bool nms_to_nms9_callback_func(pattern::Matcher& m, pass::MatcherPass* impl) { nms_9->set_friendly_name(root->get_friendly_name()); ngraph::copy_runtime_info(root, nms_9); - root->output(0).replace(nms_9->output(0)); + // nms0-4 have one output, nms5/9 have 3 outputs. + if (std::dynamic_pointer_cast(root)) + ngraph::replace_node(root, nms_9); + else + root->output(0).replace(nms_9->output(0)); return true; } } // namespace