[IE][VPU] Proposal stage: fixed check for optional 2nd output (#2980)
authorAlexey Ershov <alexey.ershov@intel.com>
Thu, 5 Nov 2020 16:05:37 +0000 (19:05 +0300)
committerGitHub <noreply@github.com>
Thu, 5 Nov 2020 16:05:37 +0000 (19:05 +0300)
Fixed check for omitted 2nd optional output

inference-engine/src/vpu/graph_transformer/src/stages/proposal.cpp

index 0b74764..b1f08c1 100644 (file)
@@ -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];