Support ppocr_mobile_det model of op11 (#272)

* fixed ppocr_det op11 has huge diff * fixed some bugs * fixed nearest bugs * fixed nearest bugs
PaddlePaddle · Jul 5, 2021 · 7893509 · 7893509
1 parent f8db7b7
commit 7893509
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/paddle2onnx/op_mapper/tensor.py b/paddle2onnx/op_mapper/tensor.py
@@ -1027,12 +1027,21 @@ def opset_11(cls, graph, node, **kw):
                 in_shape, out_shape = cls.compute_output_shape_by_size(graph,
                                                                        node)
                 inputs += [empty_node, out_shape]
-        graph.make_node(
-            'Resize',
-            inputs=inputs,
-            outputs=node.output('Out'),
-            mode=resize_type,
-            coordinate_transformation_mode=coordinate_transformation_mode)
+        if resize_type == 'nearest' and coordinate_transformation_mode == 'asymmetric':
+            graph.make_node(
+                'Resize',
+                inputs=inputs,
+                outputs=node.output('Out'),
+                mode=resize_type,
+                coordinate_transformation_mode=coordinate_transformation_mode,
+                nearest_mode='floor')
+        else:
+            graph.make_node(
+                'Resize',
+                inputs=inputs,
+                outputs=node.output('Out'),
+                mode=resize_type,
+                coordinate_transformation_mode=coordinate_transformation_mode)
 
     @classmethod
     def compute_output_shape(cls, graph, node, opset_version=10):