Skip to content

Mace
Mace

Xiaomi / Mace

通知 106
Star 40
Fork 27
Mace
Mace

体验新版 GitCode，发现更多精彩内容 >>

You need to sign in or sign up before continuing.
提交 7eb6a2b1 编写于 作者: pplly's avatar pplly
浏览文件
操作

fix caffe model converted when group == 1, is_depthwise is False.

上级 394ba003
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
mace/python/tools/converter_tool/caffe_converter.py
浏览文件 @ 7eb6a2b1
...@@ -355,7 +355,7 @@ class CaffeConverter(base_converter.ConverterInterface): ...@@ -355,7 +355,7 @@ class CaffeConverter(base_converter.ConverterInterface):
op = self.convert_general_op(caffe_op) op = self.convert_general_op(caffe_op)
param = caffe_op.layer.convolution_param param = caffe_op.layer.convolution_param
is_depthwise = False is_depthwise = False
if param.HasField(caffe_group_str): if param.HasField(caffe_group_str) and param.group > 1:
filter_data = caffe_op.blobs[0] filter_data = caffe_op.blobs[0]
mace_check(param.group == filter_data.shape[0] and mace_check(param.group == filter_data.shape[0] and
filter_data.shape[1] == 1, filter_data.shape[1] == 1,
...@@ -403,7 +403,7 @@ class CaffeConverter(base_converter.ConverterInterface): ...@@ -403,7 +403,7 @@ class CaffeConverter(base_converter.ConverterInterface):
op = self.convert_general_op(caffe_op) op = self.convert_general_op(caffe_op)
param = caffe_op.layer.convolution_param param = caffe_op.layer.convolution_param
is_depthwise = False is_depthwise = False
if param.HasField(caffe_group_str): if param.HasField(caffe_group_str) and param.group > 1:
filter_data = caffe_op.blobs[0] filter_data = caffe_op.blobs[0]
mace_check(param.group == filter_data.shape[0] and mace_check(param.group == filter_data.shape[0] and
filter_data.shape[1] == 1, filter_data.shape[1] == 1,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或