Merge pull request #17270 from l-bat:ngraph_missed_data

pull/17285/head
Alexander Alekhin 5 years ago
commit b2464e3379
  1. 2
      modules/dnn/src/layers/batch_norm_layer.cpp
  2. 2
      modules/dnn/src/layers/scale_layer.cpp
  3. 11
      modules/dnn/test/test_onnx_importer.cpp

@ -373,7 +373,7 @@ public:
shape[1] = weights_.total(); shape[1] = weights_.total();
auto weight = std::make_shared<ngraph::op::Constant>(ngraph::element::f32, ngraph::Shape(shape), weights_.data); auto weight = std::make_shared<ngraph::op::Constant>(ngraph::element::f32, ngraph::Shape(shape), weights_.data);
auto bias = std::make_shared<ngraph::op::Constant>(ngraph::element::f32, ngraph::Shape(shape), bias_.data); auto bias = std::make_shared<ngraph::op::Constant>(ngraph::element::f32, ngraph::Shape(shape), bias_.data);
auto scale_node = std::make_shared<ngraph::op::v1::Multiply>(ieInpNode, weight, ngraph::op::AutoBroadcastType::NUMPY); auto scale_node = std::make_shared<ngraph::op::v0::Multiply>(ieInpNode, weight, ngraph::op::AutoBroadcastType::NUMPY);
auto scale_shift = std::make_shared<ngraph::op::v1::Add>(scale_node, bias, ngraph::op::AutoBroadcastType::NUMPY); auto scale_shift = std::make_shared<ngraph::op::v1::Add>(scale_node, bias, ngraph::op::AutoBroadcastType::NUMPY);
return Ptr<BackendNode>(new InfEngineNgraphNode(scale_shift)); return Ptr<BackendNode>(new InfEngineNgraphNode(scale_shift));
} }

@ -248,7 +248,7 @@ public:
auto weight = blobs.empty() ? ieInpNode1 : auto weight = blobs.empty() ? ieInpNode1 :
std::make_shared<ngraph::op::Constant>(ngraph::element::f32, ngraph::Shape(shape), blobs[0].data); std::make_shared<ngraph::op::Constant>(ngraph::element::f32, ngraph::Shape(shape), blobs[0].data);
node = std::make_shared<ngraph::op::v1::Multiply>(node, weight, ngraph::op::AutoBroadcastType::NUMPY); node = std::make_shared<ngraph::op::v0::Multiply>(node, weight, ngraph::op::AutoBroadcastType::NUMPY);
} }
if (hasBias || !hasWeights) if (hasBias || !hasWeights)
{ {

@ -299,17 +299,11 @@ TEST_P(Test_ONNX_layers, BatchNormalization3D)
TEST_P(Test_ONNX_layers, BatchNormalizationUnfused) TEST_P(Test_ONNX_layers, BatchNormalizationUnfused)
{ {
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
testONNXModels("frozenBatchNorm2d"); testONNXModels("frozenBatchNorm2d");
} }
TEST_P(Test_ONNX_layers, BatchNormalizationSubgraph) TEST_P(Test_ONNX_layers, BatchNormalizationSubgraph)
{ {
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH)
applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
testONNXModels("batch_norm_subgraph"); testONNXModels("batch_norm_subgraph");
} }
@ -351,9 +345,8 @@ TEST_P(Test_ONNX_layers, Expand)
TEST_P(Test_ONNX_layers, ExpandHW) TEST_P(Test_ONNX_layers, ExpandHW)
{ {
// ngraph::op::v1::Multiply bug if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019)
if (backend == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 || backend == DNN_BACKEND_INFERENCE_ENGINE_NGRAPH) applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER);
applyTestTag(CV_TEST_TAG_DNN_SKIP_IE_NN_BUILDER, CV_TEST_TAG_DNN_SKIP_IE_NGRAPH);
testONNXModels("expand_hw"); testONNXModels("expand_hw");
} }

Loading…
Cancel
Save