Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
毕竟曾有刹那
Mace
提交
e0d6d4e7
Mace
项目概览
毕竟曾有刹那
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
1
Star
0
Fork
0
代码
文件
提交
分支
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,发现更多精彩内容 >>
提交
e0d6d4e7
编写于
9月 19, 2019
作者:
叶
叶剑武
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
avoid obfuscate tensors which are outputs and inputs of other ops at the
same time
上级
1a795788
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
2 addition
and
75 deletion
+2
-75
tools/python/encrypt.py
tools/python/encrypt.py
+2
-1
tools/sh_commands.py
tools/sh_commands.py
+0
-74
未找到文件。
tools/python/encrypt.py
浏览文件 @
e0d6d4e7
...
...
@@ -94,7 +94,8 @@ def obfuscate_name(model):
t
.
name
=
tensor_map
[
t
.
name
]
for
op
in
model
.
op
:
for
i
in
range
(
len
(
op
.
input
)):
if
op
.
input
[
i
]
not
in
input_nodes
:
if
(
op
.
input
[
i
]
not
in
input_nodes
)
and
\
(
op
.
input
[
i
]
not
in
output_nodes
):
op
.
input
[
i
]
=
in_out_map
[
op
.
input
[
i
]]
for
i
in
range
(
len
(
op
.
output
)):
if
op
.
output
[
i
]
not
in
output_nodes
:
...
...
tools/sh_commands.py
浏览文件 @
e0d6d4e7
...
...
@@ -478,80 +478,6 @@ def merge_opencl_parameters(binaries_dirs,
np
.
array
(
output_byte_array
).
tofile
(
output_file_path
)
def
gen_model_code
(
model_codegen_dir
,
platform
,
model_file_path
,
weight_file_path
,
model_sha256_checksum
,
weight_sha256_checksum
,
input_nodes
,
input_data_types
,
input_data_formats
,
output_nodes
,
output_data_types
,
output_data_formats
,
check_nodes
,
runtime
,
model_tag
,
input_shapes
,
input_ranges
,
output_shapes
,
check_shapes
,
dsp_mode
,
embed_model_data
,
winograd
,
quantize
,
quantize_large_weights
,
quantize_range_file
,
change_concat_ranges
,
obfuscate
,
model_graph_format
,
data_type
,
cl_mem_type
,
graph_optimize_options
):
bazel_build_common
(
"//mace/python/tools:converter"
)
if
os
.
path
.
exists
(
model_codegen_dir
):
sh
.
rm
(
"-rf"
,
model_codegen_dir
)
sh
.
mkdir
(
"-p"
,
model_codegen_dir
)
sh
.
python
(
"bazel-bin/mace/python/tools/converter"
,
"-u"
,
"--platform=%s"
%
platform
,
"--model_file=%s"
%
model_file_path
,
"--weight_file=%s"
%
weight_file_path
,
"--model_checksum=%s"
%
model_sha256_checksum
,
"--weight_checksum=%s"
%
weight_sha256_checksum
,
"--input_node=%s"
%
input_nodes
,
"--input_data_types=%s"
%
input_data_types
,
"--input_data_formats=%s"
%
input_data_formats
,
"--output_node=%s"
%
output_nodes
,
"--output_data_types=%s"
%
output_data_types
,
"--output_data_formats=%s"
%
output_data_formats
,
"--check_node=%s"
%
check_nodes
,
"--runtime=%s"
%
runtime
,
"--template=%s"
%
"mace/python/tools"
,
"--model_tag=%s"
%
model_tag
,
"--input_shape=%s"
%
input_shapes
,
"--input_range=%s"
%
input_ranges
,
"--output_shape=%s"
%
output_shapes
,
"--check_shape=%s"
%
check_shapes
,
"--dsp_mode=%s"
%
dsp_mode
,
"--embed_model_data=%s"
%
embed_model_data
,
"--winograd=%s"
%
winograd
,
"--quantize=%s"
%
quantize
,
"--quantize_large_weights=%s"
%
quantize_large_weights
,
"--quantize_range_file=%s"
%
quantize_range_file
,
"--change_concat_ranges=%s"
%
change_concat_ranges
,
"--obfuscate=%s"
%
obfuscate
,
"--output_dir=%s"
%
model_codegen_dir
,
"--model_graph_format=%s"
%
model_graph_format
,
"--data_type=%s"
%
data_type
,
"--graph_optimize_options=%s"
%
graph_optimize_options
,
"--cl_mem_type=%s"
%
cl_mem_type
,
_fg
=
True
)
def
gen_input
(
model_output_dir
,
input_nodes
,
input_shapes
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录