提交 bc0116dc 编写于 作者: L luxuhui 提交者: 叶剑武

fix computing error of caffe interp op

N/A
Signed-off-by: NLuxuhui <luxuhui@xiaomi.com>
上级 2b04dbe9
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册