From a2c716b9c05bce0984b59a834d8a6453c1ccfcb0 Mon Sep 17 00:00:00 2001 From: Channingss Date: Tue, 4 Aug 2020 02:56:24 +0000 Subject: [PATCH] fix bug of Conv --- x2paddle/op_mapper/onnx2paddle/opset9/opset.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/x2paddle/op_mapper/onnx2paddle/opset9/opset.py b/x2paddle/op_mapper/onnx2paddle/opset9/opset.py index 3cb60ff..2ee3aca 100644 --- a/x2paddle/op_mapper/onnx2paddle/opset9/opset.py +++ b/x2paddle/op_mapper/onnx2paddle/opset9/opset.py @@ -40,7 +40,7 @@ def _const_weight_or_none(node): return None -def get_same_padding(in_size, kernel_size, stride): +def _get_same_padding(in_size, kernel_size, stride): new_size = int(math.ceil(in_size * 1.0 / stride)) pad_size = (new_size - 1) * stride + kernel_size - in_size pad0 = int(pad_size / 2) @@ -1305,11 +1305,11 @@ class OpSet9(): paddings, val_x = self._pad_if_asymmetric(node, pads, val_x) if auto_pad == "SAME_UPPER" or auto_pad == "SAME_LOWER": - pad_h = get_same_padding(input_shape[2], kernel_shape[0], - strides[0]) - pad_w = get_same_padding(input_shape[3], kernel_shape[1], - strides[1]) - attr = {"paddings": pad_h + pad_w, "pad_value": 0.0} + pad_h = _get_same_padding(input_shape[2], kernel_shape[0], + strides[0]) + pad_w = _get_same_padding(input_shape[3], kernel_shape[1], + strides[1]) + paddings = pad_h + pad_w attr = { "num_filters": num_out_channels, -- GitLab