Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
as350144
Mace
提交
fb471e21
Mace
项目概览
as350144
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
2
Star
1
Fork
1
代码
文件
提交
分支
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,发现更多精彩内容 >>
提交
fb471e21
编写于
9月 25, 2017
作者:
L
Liangliang He
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'master' into 'master'
Fix: if reshape in BN, skip it See merge request !59
上级
a407488f
8557ed20
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
12 addition
and
7 deletion
+12
-7
mace/python/tools/tf_converter_lib.py
mace/python/tools/tf_converter_lib.py
+12
-7
未找到文件。
mace/python/tools/tf_converter_lib.py
浏览文件 @
fb471e21
...
...
@@ -12,6 +12,11 @@ pooling_type_mode = {
'MaxPool'
:
2
}
def
get_input_tensor
(
op
,
index
):
input_tensor
=
op
.
inputs
[
index
]
if
input_tensor
.
op
.
type
==
'Reshape'
:
input_tensor
=
get_input_tensor
(
input_tensor
.
op
,
0
)
return
input_tensor
def
convert_ops
(
unresolved_ops
,
net_def
):
ops_count
=
len
(
unresolved_ops
)
...
...
@@ -19,7 +24,7 @@ def convert_ops(unresolved_ops, net_def):
first_op
=
unresolved_ops
[
0
]
if
first_op
.
type
==
'Placeholder'
:
if
first_op
.
type
==
'Placeholder'
or
first_op
.
type
==
'Reshape'
:
pass
elif
first_op
.
type
==
'Const'
:
tf_tensor
=
first_op
.
outputs
[
0
].
eval
()
...
...
@@ -76,12 +81,12 @@ def convert_ops(unresolved_ops, net_def):
if
mul_op
.
type
!=
'Mul'
or
mul_2_op
.
type
!=
'Mul'
or
mul_1_op
.
type
!=
'Mul'
or
sub_op
.
type
!=
'Sub'
or
add_1_op
.
type
!=
'Add'
:
raise
Exception
(
'Invalid BatchNorm Op'
)
input_name
=
mul_1_op
.
inputs
[
0
]
.
name
gamma
=
mul_op
.
inputs
[
1
]
.
name
beta
=
sub_op
.
inputs
[
0
]
.
name
mean
=
mul_2_op
.
inputs
[
0
]
.
name
variance
=
add_op
.
inputs
[
0
]
.
name
epsilon
=
add_op
.
inputs
[
1
]
.
name
input_name
=
get_input_tensor
(
mul_1_op
,
0
)
.
name
gamma
=
get_input_tensor
(
mul_op
,
1
)
.
name
beta
=
get_input_tensor
(
sub_op
,
0
)
.
name
mean
=
get_input_tensor
(
mul_2_op
,
0
)
.
name
variance
=
get_input_tensor
(
add_op
,
0
)
.
name
epsilon
=
get_input_tensor
(
add_op
,
1
)
.
name
op_def
=
net_def
.
op
.
add
()
op_def
.
name
=
first_op
.
name
[:
-
4
]
# remove /add
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录