Merge pull request #12864 from dkurt:dnn_ie_get_batch_size

pull/12867/head
Alexander Alekhin 6 years ago
commit 6d9e66eca1
  1. 14
      modules/dnn/src/op_inf_engine.cpp

@ -331,8 +331,18 @@ InferenceEngine::StatusCode InfEngineBackendNet::setBatchSize(size_t size, Infer
size_t InfEngineBackendNet::getBatchSize() const noexcept size_t InfEngineBackendNet::getBatchSize() const noexcept
{ {
CV_Error(Error::StsNotImplemented, ""); size_t batchSize = 0;
return 0; for (const auto& inp : inputs)
{
CV_Assert(inp.second);
std::vector<size_t> dims = inp.second->getDims();
CV_Assert(!dims.empty());
if (batchSize != 0)
CV_Assert(batchSize == dims.back());
else
batchSize = dims.back();
}
return batchSize;
} }
#if INF_ENGINE_VER_MAJOR_GT(INF_ENGINE_RELEASE_2018R2) #if INF_ENGINE_VER_MAJOR_GT(INF_ENGINE_RELEASE_2018R2)

Loading…
Cancel
Save