From 868369dbf580b64303f3cf553820ab6056efb161 Mon Sep 17 00:00:00 2001 From: guosheng Date: Sun, 11 Jun 2017 17:31:26 +0800 Subject: [PATCH] update defalut name corresponding to up-to-date paddle --- image_classification/caffe2paddle/caffe2paddle.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/image_classification/caffe2paddle/caffe2paddle.py b/image_classification/caffe2paddle/caffe2paddle.py index 4d01a3c2..a4011f28 100644 --- a/image_classification/caffe2paddle/caffe2paddle.py +++ b/image_classification/caffe2paddle/caffe2paddle.py @@ -61,7 +61,7 @@ class ModelConverter(object): f.write(struct.pack("IIQ", 0, 4, data.size)) f.write(data.tobytes()) - @wrap_name_default("img_conv_layer") + @wrap_name_default("conv") def convert_Convolution_layer(self, params, name=None): for i in range(len(params)): data = np.array(params[i].data) @@ -97,7 +97,7 @@ class ModelConverter(object): self.params[file_name] = (param_conf, data.flatten()) return name - @wrap_name_default("batch_norm_layer") + @wrap_name_default("batch_norm") def convert_BatchNorm_layer(self, params, name=None): scale = 1 / np.array(params[-1].data)[0] if np.array( params[-1].data)[0] != 0 else 0 -- GitLab