Fixed windows build with InferenceEngine

pull/12339/head^2^2 3.4.3-openvino
Maksim Shabunin 7 years ago
parent 8b197c58e6
commit eff0f9d3f8
  1. 4
      modules/dnn/src/layers/normalize_bbox_layer.cpp

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

Loading…
Cancel
Save