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

Merge pull request #12864 from dkurt:dnn_ie_get_batch_size

Browse Source
This commit is contained in:
Alexander Alekhin 2018-10-17 14:02:25 +00:00
commit 6d9e66eca1
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
modules/dnn/src/op_inf_engine.cpp
View File

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