mirror of
https://github.com/opencv/opencv.git
synced 2024-11-25 03:30:34 +08:00
Limit number of threads in CPU plugin of IE by OpenCV's getNumThreads
This commit is contained in:
parent
24790e4061
commit
96a2f80bf2
@ -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);
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user