Merge pull request #952 from wjj19950828/support_yolov8cls

[Bug]Support YOLOv8-cls
PaddlePaddle · Mar 31, 2023 · 685c8ce · 685c8ce
2 parents c429515 + 96501c6
commit 685c8ce
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 724 deletions.
diff --git a/x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py b/x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py
@@ -796,11 +796,17 @@ def Scale(self, node):
                 inputs=inputs_dict,
                 outputs=[node.layer_name + "_mul"])
         else:
+            new_shape = [1] * len(node.in_shapes[0])
+            new_shape[axis] = node.in_shapes[0][1]
             self.paddle_graph.add_layer(
-                "paddle.fluid.layers.elementwise_mul",
+                "paddle.reshape",
+                inputs={"x": node.layer_name + "_cparam1"},
+                outputs=[node.layer_name + "_cparam1"],
+                shape=new_shape)
+            self.paddle_graph.add_layer(
+                "paddle.multiply",
                 inputs=inputs_dict,
-                outputs=[node.layer_name + "_mul"],
-                axis=axis)
+                outputs=[node.layer_name + "_mul"])
         self.paddle_graph.add_layer(
             "self.create_parameter",
             inputs={},