[IE][VPU] Proposal stage: fixed check for optional 2nd output (#2980)

Fixed check for omitted 2nd optional output
2020-11-05 19:05:37 +03:00
parent 022ea97f18
commit d5daf9d873
1 changed files with 1 additions and 1 deletions
--- a/inference-engine/src/vpu/graph_transformer/src/stages/proposal.cpp
+++ b/inference-engine/src/vpu/graph_transformer/src/stages/proposal.cpp
@@ -135,7 +135,7 @@ void FrontEnd::parseProposal(const Model& model, const ie::CNNLayerPtr& layer, c
    DataVector tempOutputs(2);
    tempOutputs[0] = outputs[0];

-    if (outputs.size() < 2)
+    if ((outputs.size() < 2) || !(outputs[1]))
        tempOutputs[1] = model->addFakeData();
    else
        tempOutputs[1] = outputs[1];