Merge pull request #1144 from NikolasMarkou/resize_align_corners_attribute

guschmue · web-flow · commit 57ccfb3262ae · 2020-10-24T09:34:02.000-07:00
Resize align corners attribute
diff --git a/tf2onnx/onnx_opset/nn.py b/tf2onnx/onnx_opset/nn.py
@@ -900,6 +900,8 @@ def version_11(cls, ctx, node, **kwargs):
             concat_shape.output[0]
         ]
         transformation_mode = "asymmetric"
+        if "align_corners" in node.attr and node.attr["align_corners"].i:
+            transformation_mode = "align_corners"
         if "half_pixel_centers" in node.attr and node.attr["half_pixel_centers"].i:
             transformation_mode = "half_pixel"
         resize = ctx.make_node("Resize", resize_inputs,

Original file line number	Diff line number	Diff line change
`@@ -900,6 +900,8 @@ def version_11(cls, ctx, node, **kwargs):`
`900`	`900`	`concat_shape.output[0]`
`901`	`901`	`]`
`902`	`902`	`transformation_mode = "asymmetric"`
	`903`	`+ if "align_corners" in node.attr and node.attr["align_corners"].i:`
	`904`	`+ transformation_mode = "align_corners"`
`903`	`905`	`if "half_pixel_centers" in node.attr and node.attr["half_pixel_centers"].i:`
`904`	`906`	`transformation_mode = "half_pixel"`
`905`	`907`	`resize = ctx.make_node("Resize", resize_inputs,`