From: Андрей Шедько/AI Tools Lab /SRR/Assistant Engineer/삼성전자 <a.shedko@partner.samsung.com>
Date: Mon, 17 Sep 2018 14:37:15 +0000 (+0300)
Subject: Fix Importer bug (#1541)
X-Git-Tag: nncc_backup~1785
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=e151c351931d8341ac72da03a630a1d88d45f482;p=platform%2Fcore%2Fml%2Fnnfw.git

Fix Importer bug (#1541)

Found when testing 1 op models: `Im2Col` Layer uses the convolution parameter, as does
the `Deconvolution` (or `TransposedConvolution` layer). This resolves the issue.

Signed-off-by: Andrei Shedko <a.shedko@partner.samsung.com>
---

diff --git a/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp b/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp
index 5083140..bc93ce0 100644
--- a/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp
+++ b/contrib/nnc/passes/caffe_frontend/caffe_model_visitor.cpp
@@ -43,7 +43,7 @@ void ModelVisitor::visit(const LayerParameter& lp)
   {
     processInputLayer(lp);
   }
-  else if (lp.has_convolution_param())
+  else if (lp.type() == "Convolution")
   {
     outputs = opCreator.createConv2D(inputs, params, lp.convolution_param());
   }