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

Set dontReverseIndices to Gather if its indices come from NMS v9 (#12478)

Browse Source 
Ticket: 86909
This commit is contained in:
Mateusz Tabaka 2022-08-10 11:31:39 +02:00 committed by GitHub
parent 6e4a8b0d2e
commit c328db5aac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/bindings/python/tests/__init__.py
View File

@ -131,5 +131,4 @@ xfail_issue_82038 = xfail_test(reason="ScatterElements, ScatterND, AssertionErro
xfail_issue_82039 = xfail_test(reason="Unsupported data type Optional, RuntimeError: [ NOT_IMPLEMENTED ] "
"CPU plugin: Input image format UNSPECIFIED is not supported yet...")
xfail_issue_86909 = xfail_test(reason="ssd_resnet34 - AssertionError: zoo models results mismatch")
xfail_issue_86911 = xfail_test(reason="LSTM_Seq_len_unpacked - AssertionError: zoo models results mismatch")

2
src/bindings/python/tests/test_onnx/test_zoo_models.py
View File

@ -21,7 +21,6 @@ from tests import (
xfail_issue_48190,
xfail_issue_58676,
xfail_issue_78843,
xfail_issue_86909,
xfail_issue_86911,
xfail_issue_onnx_models_140)
@ -194,7 +193,6 @@ if len(zoo_models) > 0:
(xfail_issue_47495, "test_MSFT_opset10_BERT_Squad_bertsquad10_cpu"),
(xfail_issue_78843, "test_MSFT_opset10_mlperf_ssd_mobilenet_300_ssd_mobilenet_v1_coco_2018_01_28_cpu"),
(xfail_issue_86909, "test_MSFT_opset10_mlperf_ssd_resnet34_1200_ssd_resnet34_mAP_20.2_cpu"),
(xfail_issue_86911, "test_MSFT_opset9_LSTM_Seq_lens_unpacked_model_cpu"),
]

7
src/common/transformations/src/transformations/common_optimizations/convert_nms_gather_path_to_unsigned.cpp
View File

@ -10,6 +10,7 @@
#include <ngraph/opsets/opset3.hpp>
#include <ngraph/opsets/opset5.hpp>
#include <ngraph/opsets/opset8.hpp>
#include <ngraph/opsets/opset9.hpp>
#include <ngraph/pattern/op/wrap_type.hpp>
#include <ngraph/rt_info.hpp>
#include <transformations/rt_info/nms_selected_indices.hpp>
@ -26,8 +27,10 @@ public:
OPENVINO_RTTI("InitNMSPath", "0");
InitNMSPath() {
MATCHER_SCOPE(InitNMSPath);
auto nms_pattern =
pattern::wrap_type<opset1::NonMaxSuppression, opset3::NonMaxSuppression, opset5::NonMaxSuppression>();
auto nms_pattern = pattern::wrap_type<opset1::NonMaxSuppression,
opset3::NonMaxSuppression,
opset5::NonMaxSuppression,
opset9::NonMaxSuppression>();
matcher_pass_callback callback = [=](pattern::Matcher& m) {
const auto& out_nodes = m.get_match_root()->output(0).get_target_inputs();
for (const auto& out_node : out_nodes) {