Skip to content

O
opencv
Commit c9faa09d authored by Alexander Alekhin's avatar Alexander Alekhin
Browse files
Options

Merge pull request #12266 from mshabunin:fix-windows-ie-build

parents 1deeca98 808c89ad
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
modules/dnn/src/layers/normalize_bbox_layer.cpp
View file @ c9faa09d
...@@ -276,7 +276,7 @@ public: ...@@ -276,7 +276,7 @@ public:
{ {
auto weights = InferenceEngine::make_shared_blob<float>(InferenceEngine::Precision::FP32, auto weights = InferenceEngine::make_shared_blob<float>(InferenceEngine::Precision::FP32,
InferenceEngine::Layout::C, InferenceEngine::Layout::C,
{numChannels}); {(size_t)numChannels});
weights->allocate(); weights->allocate();
std::vector<float> ones(numChannels, 1); std::vector<float> ones(numChannels, 1);
weights->set(ones); weights->set(ones);
...@@ -286,7 +286,7 @@ public: ...@@ -286,7 +286,7 @@ public:
else else
{ {
CV_Assert(numChannels == blobs[0].total()); CV_Assert(numChannels == blobs[0].total());
ieLayer->blobs["weights"] = wrapToInfEngineBlob(blobs[0], {numChannels}, InferenceEngine::Layout::C); ieLayer->blobs["weights"] = wrapToInfEngineBlob(blobs[0], {(size_t)numChannels}, InferenceEngine::Layout::C);
ieLayer->params["channel_shared"] = blobs[0].total() == 1 ? "1" : "0"; ieLayer->params["channel_shared"] = blobs[0].total() == 1 ? "1" : "0";
} }
ieLayer->params["eps"] = format("%f", epsilon); ieLayer->params["eps"] = format("%f", epsilon);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment