Limit number of threads in CPU plugin of IE by OpenCV's getNumThreads

This commit is contained in:
Dmitry Kurtaev 2019-06-24 22:41:30 +03:00
parent 24790e4061
commit 96a2f80bf2

View File

@ -460,6 +460,12 @@ void InfEngineBackendNet::initPlugin(InferenceEngine::ICNNNetwork& net)
CV_LOG_WARNING(NULL, "DNN-IE: Can't load extension plugin (extra layers for some networks). Specify path via OPENCV_DNN_IE_EXTRA_PLUGIN_PATH parameter");
}
// Some of networks can work without a library of extra layers.
#ifndef _WIN32
// Limit the number of CPU threads.
enginePtr->SetConfig({{
InferenceEngine::PluginConfigParams::KEY_CPU_THREADS_NUM, format("%d", getNumThreads()),
}}, 0);
#endif
}
plugin = InferenceEngine::InferencePlugin(enginePtr);