From 35b0c3ed6bf6d0ead4a178f4f6dbb277d145e0fe Mon Sep 17 00:00:00 2001 From: channingss Date: Tue, 13 Aug 2019 13:29:20 +0800 Subject: [PATCH] support model Face_cyclegan --- x2paddle/op_mapper/onnx_op_mapper.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/x2paddle/op_mapper/onnx_op_mapper.py b/x2paddle/op_mapper/onnx_op_mapper.py index c7c113c..675ca96 100644 --- a/x2paddle/op_mapper/onnx_op_mapper.py +++ b/x2paddle/op_mapper/onnx_op_mapper.py @@ -823,14 +823,13 @@ class ONNXOpMapper(OpMapper): paddings, var_x = self._pad_if_asymmetric(node, pads, val_x) output_size = [0, 0] - print(val_x.out_shapes[0]) + output_size[0] = (val_x.out_shapes[0][2] - 1) * strides[0] - 2 * paddings[0] + dilations[0] * ( kernel_shape[0] - 1) + 1 + out_padding[0] output_size[1] = (val_x.out_shapes[0][3] - 1) * strides[1] - 2 * paddings[1] + dilations[1] * ( kernel_shape[1] - 1) + 1 + out_padding[1] - print(output_size) attr = { 'num_filters': num_out_channels, 'output_size': output_size or None, -- GitLab