IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE][VPU]: ignore 2nd output of Proposal layer (#1975)

Browse Source 
* TODO: implement 2nd output, see: #-37327: Several models Failed to compile layer "proposals"
This commit is contained in:
Evgeny Latkin 2020-08-28 17:18:42 +03:00 committed by GitHub
parent 990f4e2919
commit 245920a95d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
inference-engine/src/vpu/graph_transformer/src/stages/proposal.cpp
View File

@ -123,9 +123,13 @@ void FrontEnd::parseProposal(const Model& model, const ie::CNNLayerPtr& layer, c
ie::details::CaselessEq<std::string> cmp;
IE_ASSERT(inputs.size() == 3);
IE_ASSERT(outputs.size() == 1);
auto stage = model->addNewStage<ProposalStage>(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<ProposalStage>(layer->name, StageType::Proposal, layer, inputs, outputs1);
stage->attrs().set<int>("feat_stride", layer->GetParamAsInt("feat_stride", 16));
stage->attrs().set<int>("base_size", layer->GetParamAsInt("base_size", 16));