From be86338a79d7433860a3dfa1e40b0806d662a54e Mon Sep 17 00:00:00 2001 From: Liubov Batanina Date: Tue, 14 Jan 2020 12:51:19 +0300 Subject: [PATCH] Enable acrossSpatial normalizeL2 on Myriad --- modules/dnn/src/layers/normalize_bbox_layer.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/dnn/src/layers/normalize_bbox_layer.cpp b/modules/dnn/src/layers/normalize_bbox_layer.cpp index 63f93e2eb5..6c1d381feb 100644 --- a/modules/dnn/src/layers/normalize_bbox_layer.cpp +++ b/modules/dnn/src/layers/normalize_bbox_layer.cpp @@ -69,7 +69,10 @@ public: if (pnorm != 2) return false; - return preferableTarget == DNN_TARGET_MYRIAD ? !acrossSpatial : startAxis == 1; + if (backendId == DNN_BACKEND_INFERENCE_ENGINE_NN_BUILDER_2019 && preferableTarget == DNN_TARGET_MYRIAD) + return !acrossSpatial; + + return startAxis == 1; } return backendId == DNN_BACKEND_OPENCV; } @@ -339,7 +342,6 @@ public: } else { - // weight->get_shape().size() > 1 ~> channel_shared = false weight = std::make_shared( ngraph::element::f32, ngraph::Shape(shape), blobs[0].data); }