From bc0116dc48296038f039a448577f87620d67ecb7 Mon Sep 17 00:00:00 2001 From: luxuhui Date: Fri, 27 Sep 2019 12:22:55 +0800 Subject: [PATCH] fix computing error of caffe interp op N/A Signed-off-by: Luxuhui --- tools/python/transform/caffe_converter.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/python/transform/caffe_converter.py b/tools/python/transform/caffe_converter.py index a185755b..fd75501b 100644 --- a/tools/python/transform/caffe_converter.py +++ b/tools/python/transform/caffe_converter.py @@ -610,6 +610,10 @@ class CaffeConverter(base_converter.ConverterInterface): size_arg.name = MaceKeyword.mace_resize_size_str size_value = np.array([param.height, param.width], dtype=np.int32) size_arg.ints.extend(size_value) + # interp op's `align_corners` param is always true in caffe + align_corners_arg = op.arg.add() + align_corners_arg.name = MaceKeyword.mace_align_corners_str + align_corners_arg.i = 1 def convert_fully_connected(self, caffe_op): op = self.convert_general_op(caffe_op) -- GitLab