From d5daf9d873fa490ac8151a287984be0ce90ba72a Mon Sep 17 00:00:00 2001 From: Alexey Ershov Date: Thu, 5 Nov 2020 19:05:37 +0300 Subject: [PATCH] [IE][VPU] Proposal stage: fixed check for optional 2nd output (#2980) Fixed check for omitted 2nd optional output --- .../src/vpu/graph_transformer/src/stages/proposal.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/inference-engine/src/vpu/graph_transformer/src/stages/proposal.cpp b/inference-engine/src/vpu/graph_transformer/src/stages/proposal.cpp index 0b74764bb91..b1f08c1d8d3 100644 --- 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];