Added align_corners for Resize OPSET 11

Signed-off-by: Nikolas Markou <nikolasmarkou@gmail.com>
onnx · Oct 23, 2020 · cdf67c0 · cdf67c0
1 parent de21e8a
commit cdf67c0
Showing 1 changed file with 2 additions and 0 deletions.
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,