From 96a2f80bf28a7ad8b7c8973b581f67bc2c143640 Mon Sep 17 00:00:00 2001 From: Dmitry Kurtaev Date: Mon, 24 Jun 2019 22:41:30 +0300 Subject: [PATCH] Limit number of threads in CPU plugin of IE by OpenCV's getNumThreads --- modules/dnn/src/op_inf_engine.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/modules/dnn/src/op_inf_engine.cpp b/modules/dnn/src/op_inf_engine.cpp index 3f96a1db37..ab9bdc38fe 100644 --- a/modules/dnn/src/op_inf_engine.cpp +++ b/modules/dnn/src/op_inf_engine.cpp @@ -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);