IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix_nms_ops_transformation (#11794)

Browse Source 
* fix_nms_ops_transformation

* replace node when 5-9
This commit is contained in:
Chenhu Wang 2022-06-10 10:18:28 +08:00 committed by GitHub
parent 604dc4589c
commit 1066d4551f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/common/transformations/src/transformations/op_conversions/convert_previous_nms_to_nms_9.cpp
View File

@ -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<ngraph::opset5::NonMaxSuppression>(root))
ngraph::replace_node(root, nms_9);
else
root->output(0).replace(nms_9->output(0));
return true;
}
} // namespace