未验证 提交 8e49a687 编写于 作者: J Jason 提交者: GitHub

Merge pull request #373 from Channingss/fix_nearest

Fix nearest
...@@ -196,7 +196,7 @@ class OpSet11(OpSet10): ...@@ -196,7 +196,7 @@ class OpSet11(OpSet10):
if align_corners: if align_corners:
coordinate_transformation_mode = 'align_corners' coordinate_transformation_mode = 'align_corners'
else: else:
coordinate_transformation_mode = 'asymmetric' coordinate_transformation_mode = 'half_pixel'
roi_name = self.get_name(op.type, 'roi') roi_name = self.get_name(op.type, 'roi')
roi_node = self.make_constant_node(roi_name, onnx_pb.TensorProto.FLOAT, roi_node = self.make_constant_node(roi_name, onnx_pb.TensorProto.FLOAT,
[1, 1, 1, 1, 1, 1, 1, 1]) [1, 1, 1, 1, 1, 1, 1, 1])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册