Exclude Input layer from list of outputs for IE networks

This commit is contained in:
Dmitry Kurtaev
2018-12-06 09:12:05 +03:00
parent 09b3dcb6db
commit 93971a53d9

View File

@@ -424,6 +424,8 @@ void InfEngineBackendNet::init(int targetId)
InferenceEngine::OutputsDataMap unconnectedOuts; InferenceEngine::OutputsDataMap unconnectedOuts;
for (const auto& l : layers) for (const auto& l : layers)
{ {
if (l->type == "Input")
continue;
// Add all outputs. // Add all outputs.
for (const InferenceEngine::DataPtr& out : l->outData) for (const InferenceEngine::DataPtr& out : l->outData)
{ {