Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Xiaomi
Mace
提交
0ecfd7a0
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,发现更多精彩内容 >>
提交
0ecfd7a0
编写于
4月 26, 2018
作者:
L
Liangliang He
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add model weights checksum for caffe model
上级
1a86f7b8
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
10 addition
and
6 deletion
+10
-6
mace/python/tools/converter.py
mace/python/tools/converter.py
+3
-2
mace/python/tools/model.jinja2
mace/python/tools/model.jinja2
+1
-1
mace/python/tools/source_converter_lib.py
mace/python/tools/source_converter_lib.py
+6
-3
未找到文件。
mace/python/tools/converter.py
浏览文件 @
0ecfd7a0
...
...
@@ -45,6 +45,7 @@ def main(unused_args):
FLAGS
.
model_checksum
))
sys
.
exit
(
-
1
)
weight_checksum
=
None
if
FLAGS
.
platform
==
'caffe'
:
if
not
os
.
path
.
isfile
(
FLAGS
.
weight_file
):
print
(
"Input weight file '"
+
FLAGS
.
weight_file
+
...
...
@@ -82,8 +83,8 @@ def main(unused_args):
if
FLAGS
.
output_type
==
'source'
:
source_converter_lib
.
convert_to_source
(
output_graph_def
,
model_checksum
,
FLAGS
.
template
,
FLAGS
.
obfusc
ate
,
FLAGS
.
model_tag
,
FLAGS
.
output
,
FLAGS
.
runtime
,
output_graph_def
,
model_checksum
,
weight_checksum
,
FLAGS
.
templ
ate
,
FLAGS
.
obfuscate
,
FLAGS
.
model_tag
,
FLAGS
.
output
,
FLAGS
.
runtime
,
FLAGS
.
embed_model_data
,
FLAGS
.
winograd
)
else
:
with
open
(
FLAGS
.
output
,
"wb"
)
as
f
:
...
...
mace/python/tools/model.jinja2
浏览文件 @
0ecfd7a0
...
...
@@ -159,7 +159,7 @@ const std::string ModelName() {
}
const std::string ModelChecksum() {
return {{
model_pb_
checksum|tojson }};
return {{ checksum|tojson }};
}
const std::string ModelBuildTime() {
...
...
mace/python/tools/source_converter_lib.py
浏览文件 @
0ecfd7a0
...
...
@@ -124,8 +124,8 @@ def stringfy(value):
return
', '
.
join
(
'"{0}"'
.
format
(
w
)
for
w
in
value
)
def
convert_to_source
(
net_def
,
mode
_pb_checksum
,
template_dir
,
obfuscate
,
model_tag
,
output
,
runtime
,
embed_model_data
,
def
convert_to_source
(
net_def
,
mode
l_checksum
,
weight_checksum
,
template_dir
,
obfuscate
,
model_tag
,
output
,
runtime
,
embed_model_data
,
winograd_conv
):
if
obfuscate
:
obfuscate_name
(
net_def
)
...
...
@@ -201,6 +201,9 @@ def convert_to_source(net_def, mode_pb_checksum, template_dir, obfuscate,
TensorInfo
(
i
,
net_def
.
tensors
[
i
],
runtime
)
for
i
in
range
(
len
(
net_def
.
tensors
))
]
checksum
=
model_checksum
if
weight_checksum
is
not
None
:
checksum
=
"{},{}"
.
format
(
model_checksum
,
weight_checksum
)
source
=
j2_env
.
get_template
(
template_name
).
render
(
tensors
=
tensors
,
net
=
net_def
,
...
...
@@ -209,7 +212,7 @@ def convert_to_source(net_def, mode_pb_checksum, template_dir, obfuscate,
obfuscate
=
obfuscate
,
embed_model_data
=
embed_model_data
,
winograd_conv
=
winograd_conv
,
model_pb_checksum
=
mode_pb_
checksum
,
checksum
=
checksum
,
build_time
=
build_time
)
with
open
(
output
,
"wb"
)
as
f
:
f
.
write
(
source
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录