From 476de095cac6dd8cf163ce5d63b25ce69b229ee4 Mon Sep 17 00:00:00 2001 From: yejianwu Date: Wed, 22 Aug 2018 10:11:22 +0800 Subject: [PATCH] fix slice converter for caffe --- mace/python/tools/converter_tool/caffe_converter.py | 2 +- mace/python/tools/converter_tool/shape_inference.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mace/python/tools/converter_tool/caffe_converter.py b/mace/python/tools/converter_tool/caffe_converter.py index 5855725d..d9cc35aa 100644 --- a/mace/python/tools/converter_tool/caffe_converter.py +++ b/mace/python/tools/converter_tool/caffe_converter.py @@ -561,7 +561,7 @@ class CaffeConverter(base_converter.ConverterInterface): def convert_slice(self, caffe_op): op = self.convert_general_op(caffe_op) - op.type = MaceOp.Slice.name + op.type = MaceOp.Split.name if caffe_op.layer.HasField('slice_param'): param = caffe_op.layer.slice_param diff --git a/mace/python/tools/converter_tool/shape_inference.py b/mace/python/tools/converter_tool/shape_inference.py index 3861d648..463a5117 100644 --- a/mace/python/tools/converter_tool/shape_inference.py +++ b/mace/python/tools/converter_tool/shape_inference.py @@ -40,7 +40,7 @@ class ShapeInference(object): MaceOp.Activation.name: self.infer_shape_general, MaceOp.Pooling.name: self.infer_shape_conv_pool_shape, MaceOp.Concat.name: self.infer_shape_concat, - MaceOp.Slice.name: self.infer_shape_slice, + MaceOp.Split.name: self.infer_shape_slice, MaceOp.Softmax.name: self.infer_shape_general, MaceOp.FullyConnected.name: self.infer_shape_fully_connected, MaceOp.Crop.name: self.infer_shape_crop, -- GitLab