Fix ENet test with OpenVINO 2020.2
authorLiubov Batanina <piccione-mail@yandex.ru>
Fri, 3 Apr 2020 12:40:11 +0000 (15:40 +0300)
committerLiubov Batanina <piccione-mail@yandex.ru>
Mon, 6 Apr 2020 14:29:43 +0000 (17:29 +0300)
modules/dnn/src/dnn.cpp

index 0f8daca..f33bbbd 100644 (file)
@@ -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))