Merge pull request #20345 from mitruska:update_ngraph_normalizel2

pull/20347/head^2
Alexander Alekhin 4 years ago
commit 0f2f966a91
  1. 2
      modules/dnn/src/layers/normalize_bbox_layer.cpp

@ -328,7 +328,7 @@ public:
std::iota(axes_data.begin(), axes_data.end(), 1); std::iota(axes_data.begin(), axes_data.end(), 1);
} }
auto axes = std::make_shared<ngraph::op::Constant>(ngraph::element::i64, ngraph::Shape{axes_data.size()}, axes_data); auto axes = std::make_shared<ngraph::op::Constant>(ngraph::element::i64, ngraph::Shape{axes_data.size()}, axes_data);
auto norm = std::make_shared<ngraph::op::NormalizeL2>(ieInpNode, axes, epsilon, ngraph::op::EpsMode::ADD); auto norm = std::make_shared<ngraph::op::v0::NormalizeL2>(ieInpNode, axes, epsilon, ngraph::op::EpsMode::ADD);
CV_Assert(blobs.empty() || numChannels == blobs[0].total()); CV_Assert(blobs.empty() || numChannels == blobs[0].total());
std::vector<size_t> shape(ieInpNode->get_shape().size(), 1); std::vector<size_t> shape(ieInpNode->get_shape().size(), 1);

Loading…
Cancel
Save