提交 596a0125 编写于 作者: A Alexander Alekhin

Merge pull request #12336 from dkurt:dnn_ie_fix_net_lifetime

...@@ -161,6 +161,7 @@ InfEngineBackendNet::InfEngineBackendNet(InferenceEngine::CNNNetwork& net) ...@@ -161,6 +161,7 @@ InfEngineBackendNet::InfEngineBackendNet(InferenceEngine::CNNNetwork& net)
inputs = net.getInputsInfo(); inputs = net.getInputsInfo();
outputs = net.getOutputsInfo(); outputs = net.getOutputsInfo();
layers.resize(net.layerCount()); // A hack to execute InfEngineBackendNet::layerCount correctly. layers.resize(net.layerCount()); // A hack to execute InfEngineBackendNet::layerCount correctly.
netOwner = net;
} }
void InfEngineBackendNet::Release() noexcept void InfEngineBackendNet::Release() noexcept
......
...@@ -131,6 +131,8 @@ private: ...@@ -131,6 +131,8 @@ private:
InferenceEngine::InferencePlugin plugin; InferenceEngine::InferencePlugin plugin;
InferenceEngine::ExecutableNetwork netExec; InferenceEngine::ExecutableNetwork netExec;
InferenceEngine::InferRequest infRequest; InferenceEngine::InferRequest infRequest;
// In case of models from Model Optimizer we need to manage their lifetime.
InferenceEngine::CNNNetwork netOwner;
std::string name; std::string name;
......
...@@ -177,10 +177,6 @@ TEST_P(DNNTestOpenVINO, models) ...@@ -177,10 +177,6 @@ TEST_P(DNNTestOpenVINO, models)
Target target = (dnn::Target)(int)get<0>(GetParam()); Target target = (dnn::Target)(int)get<0>(GetParam());
std::string modelName = get<1>(GetParam()); std::string modelName = get<1>(GetParam());
if ((modelName == "semantic-segmentation-adas-0001" && target == DNN_TARGET_OPENCL_FP16) ||
(modelName == "vehicle-license-plate-detection-barrier-0106"))
throw SkipTestException("");
std::string precision = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? "FP16" : "FP32"; std::string precision = (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) ? "FP16" : "FP32";
std::string prefix = utils::fs::join("intel_models", std::string prefix = utils::fs::join("intel_models",
utils::fs::join(modelName, utils::fs::join(modelName,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册