Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Xiaomi
Mace
提交
05dfbab2
Mace
项目概览
Xiaomi
/
Mace
通知
106
Star
40
Fork
27
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Mace
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
05dfbab2
编写于
5月 16, 2018
作者:
李
李寅
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Make converter compitable with caffe of old version
上级
1d0f9cc1
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
9 addition
and
7 deletion
+9
-7
mace/python/tools/converter_tool/caffe_converter.py
mace/python/tools/converter_tool/caffe_converter.py
+9
-7
未找到文件。
mace/python/tools/converter_tool/caffe_converter.py
浏览文件 @
05dfbab2
...
@@ -374,7 +374,9 @@ class CaffeConverter(base_converter.ConverterInterface):
...
@@ -374,7 +374,9 @@ class CaffeConverter(base_converter.ConverterInterface):
if
len
(
caffe_op
.
blobs
)
==
2
:
if
len
(
caffe_op
.
blobs
)
==
2
:
bias_tensor_name
=
op
.
name
+
'_bias'
bias_tensor_name
=
op
.
name
+
'_bias'
bias_data
=
caffe_op
.
blobs
[
1
]
bias_data
=
caffe_op
.
blobs
[
1
]
self
.
add_tensor
(
bias_tensor_name
,
bias_data
.
shape
,
# caffe of old version has 4-dimension bias, so reshape it
# to single dimension
self
.
add_tensor
(
bias_tensor_name
,
bias_data
.
reshape
(
-
1
).
shape
,
mace_pb2
.
DT_FLOAT
,
mace_pb2
.
DT_FLOAT
,
bias_data
)
bias_data
)
op
.
input
.
extend
([
bias_tensor_name
])
op
.
input
.
extend
([
bias_tensor_name
])
...
@@ -407,7 +409,7 @@ class CaffeConverter(base_converter.ConverterInterface):
...
@@ -407,7 +409,7 @@ class CaffeConverter(base_converter.ConverterInterface):
if
caffe_op
.
type
==
'PReLU'
:
if
caffe_op
.
type
==
'PReLU'
:
alpha_tensor_name
=
caffe_op
.
name
+
'_alpha'
alpha_tensor_name
=
caffe_op
.
name
+
'_alpha'
alpha_data
=
caffe_op
.
blobs
[
0
]
alpha_data
=
caffe_op
.
blobs
[
0
]
self
.
add_tensor
(
alpha_tensor_name
,
alpha_data
.
shape
,
self
.
add_tensor
(
alpha_tensor_name
,
alpha_data
.
reshape
(
-
1
).
shape
,
mace_pb2
.
DT_FLOAT
,
alpha_data
)
mace_pb2
.
DT_FLOAT
,
alpha_data
)
op
.
input
.
extend
([
alpha_tensor_name
])
op
.
input
.
extend
([
alpha_tensor_name
])
...
@@ -437,10 +439,10 @@ class CaffeConverter(base_converter.ConverterInterface):
...
@@ -437,10 +439,10 @@ class CaffeConverter(base_converter.ConverterInterface):
offset_value
=
((
-
mean_value
*
scale_value
)
+
beta_value
).
reshape
(
-
1
)
offset_value
=
((
-
mean_value
*
scale_value
)
+
beta_value
).
reshape
(
-
1
)
input_names
=
[
op
.
name
+
'_scale'
,
op
.
name
+
'_offset'
]
input_names
=
[
op
.
name
+
'_scale'
,
op
.
name
+
'_offset'
]
self
.
add_tensor
(
input_names
[
0
],
scale_value
.
shape
,
mace_pb2
.
DT_FLOAT
,
self
.
add_tensor
(
input_names
[
0
],
scale_value
.
reshape
(
-
1
).
shape
,
scale_value
)
mace_pb2
.
DT_FLOAT
,
scale_value
)
self
.
add_tensor
(
input_names
[
1
],
offset_value
.
shape
,
mace_pb2
.
DT_FLOAT
,
self
.
add_tensor
(
input_names
[
1
],
offset_value
.
reshape
(
-
1
).
shape
,
offset_value
)
mace_pb2
.
DT_FLOAT
,
offset_value
)
op
.
input
.
extend
([
name
for
name
in
input_names
])
op
.
input
.
extend
([
name
for
name
in
input_names
])
op
.
output
[:]
=
scale_op
.
layer
.
top
[:]
op
.
output
[:]
=
scale_op
.
layer
.
top
[:]
...
@@ -508,7 +510,7 @@ class CaffeConverter(base_converter.ConverterInterface):
...
@@ -508,7 +510,7 @@ class CaffeConverter(base_converter.ConverterInterface):
if
len
(
caffe_op
.
blobs
)
==
2
:
if
len
(
caffe_op
.
blobs
)
==
2
:
bias_tensor_name
=
op
.
name
+
'_bias'
bias_tensor_name
=
op
.
name
+
'_bias'
bias_data
=
caffe_op
.
blobs
[
1
]
bias_data
=
caffe_op
.
blobs
[
1
]
self
.
add_tensor
(
bias_tensor_name
,
bias_data
.
shape
,
self
.
add_tensor
(
bias_tensor_name
,
bias_data
.
reshape
(
-
1
).
shape
,
mace_pb2
.
DT_FLOAT
,
mace_pb2
.
DT_FLOAT
,
bias_data
)
bias_data
)
op
.
input
.
extend
([
bias_tensor_name
])
op
.
input
.
extend
([
bias_tensor_name
])
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录