mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-05 22:19:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #16344 from l-bat:ngraph_normalize

Browse Source
This commit is contained in:
Alexander Alekhin 2020-01-14 11:55:37 +00:00
commit abe2ca886d
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/dnn/src/layers/normalize_bbox_layer.cpp
View File

@ -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::op::Constant>(
ngraph::element::f32, ngraph::Shape(shape), blobs[0].data);
}