[nnc] Fixed caffe2 maxpool border type (#2727)
authorIvan Vagin/AI Tools Lab /SRR/Engineer/삼성전자 <ivan.vagin@samsung.com>
Wed, 19 Dec 2018 15:30:44 +0000 (18:30 +0300)
committerРоман Михайлович Русяев/AI Tools Lab /SRR/Staff Engineer/삼성전자 <r.rusyaev@samsung.com>
Wed, 19 Dec 2018 15:30:44 +0000 (18:30 +0300)
Fixed caffe2 importer maxpool border type

Signed-off-by: Ivan Vagin <ivan.vagin@samsung.com>
contrib/nnc/driver/Options.cpp
contrib/nnc/passes/caffe2_frontend/caffe2_op_creator.cpp

index 0fc613e..889ca81 100644 (file)
@@ -60,7 +60,7 @@ Option<bool>  onnxFrontend(optname("--onnx"),
 Option<bool> caffe2Frontend(optname("--caffe2"),
                             overview("treat input file as Caffe2 model (predict_net.pb)"),
                             false,
-                            optional(true),
+                            optional(false),
                             optvalues(""),
                             nullptr,
                             separators(""),
index 8a679a9..478ec6b 100644 (file)
@@ -311,7 +311,7 @@ std::vector<IODescriptor> Caffe2OpCreator::convertMaxPool(const std::vector<IODe
   Shape strides = Shape({stride, stride});
 
   ops::PoolOp::PoolingType pool_type = ops::PoolOp::PoolingType::MAX;
-  ops::PoolOp::BorderType border_type = ops::PoolOp::BorderType::ZEROFILLED;
+  ops::PoolOp::BorderType border_type = ops::PoolOp::BorderType::EMPTY;
 
   std::vector<int32_t> pad_before, pad_after;
   std::tie(pad_before, pad_after) = getPadding(op);