From 245920a95d4d31b8777d3648f1456427cf4fb14a Mon Sep 17 00:00:00 2001 From: Evgeny Latkin Date: Fri, 28 Aug 2020 17:18:42 +0300 Subject: [PATCH] [IE][VPU]: ignore 2nd output of Proposal layer (#1975) * TODO: implement 2nd output, see: #-37327: Several models Failed to compile layer "proposals" --- .../src/vpu/graph_transformer/src/stages/proposal.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) 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 27d2d73..f41a363 100644 --- a/inference-engine/src/vpu/graph_transformer/src/stages/proposal.cpp +++ b/inference-engine/src/vpu/graph_transformer/src/stages/proposal.cpp @@ -123,9 +123,13 @@ void FrontEnd::parseProposal(const Model& model, const ie::CNNLayerPtr& layer, c ie::details::CaselessEq cmp; IE_ASSERT(inputs.size() == 3); - IE_ASSERT(outputs.size() == 1); - auto stage = model->addNewStage(layer->name, StageType::Proposal, layer, inputs, outputs); + // TODO: implement 2nd output, see: + // #-37327: Several models Failed to compile layer "proposals" + IE_ASSERT(outputs.size() == 1 || outputs.size() == 2); + + const DataVector outputs1 = { outputs[0] }; // ignore 2nd output + auto stage = model->addNewStage(layer->name, StageType::Proposal, layer, inputs, outputs1); stage->attrs().set("feat_stride", layer->GetParamAsInt("feat_stride", 16)); stage->attrs().set("base_size", layer->GetParamAsInt("base_size", 16)); -- 2.7.4