Fix ENet test with OpenVINO 2020.2

a448d3a6 · Liubov Batanina · 396f43d6 · a448d3a6
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

modules/dnn/src/dnn.cpp modules/dnn/src/dnn.cpp +4 -0

未找到文件。
--- a/modules/dnn/src/dnn.cpp
+++ b/modules/dnn/src/dnn.cpp
@@ -2100,7 +2100,11 @@ struct Net::Impl

                    auto ieInpNode = inputNodes[i].dynamicCast<InfEngineNgraphNode>();
                    CV_Assert(oid < ieInpNode->node->get_output_size());
+#if INF_ENGINE_VER_MAJOR_GT(2020020000)
+                    inputNodes[i] = Ptr<BackendNode>(new InfEngineNgraphNode(ieInpNode->node->get_output_as_single_output_node(oid)));
+#else
                    inputNodes[i] = Ptr<BackendNode>(new InfEngineNgraphNode(ieInpNode->node->get_output_as_single_output_node(oid, false)));
+#endif
                }

                if (layer->supportBackend(preferableBackend))