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

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

Browse Source
This commit is contained in:
Alexander Alekhin 2018-08-21 13:07:44 +00:00
commit c9faa09d55
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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:
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);