Added align_corners for Resize OPSET 11

NikolasMarkou · NikolasMarkou · commit cdf67c09506d · 2020-10-23T09:13:32.000+03:00
Signed-off-by: Nikolas Markou &lt;nikolasmarkou@gmail.com&gt;
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,`