Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Xiaomi
Mace
提交
c723f6f4
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,发现更多精彩内容 >>
提交
c723f6f4
编写于
4月 23, 2018
作者:
Y
yejianwu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix common node name conflit when validate multi tf model
上级
f2fc2197
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
6 addition
and
4 deletion
+6
-4
tools/sh_commands.py
tools/sh_commands.py
+5
-4
tools/validate.py
tools/validate.py
+1
-0
未找到文件。
tools/sh_commands.py
浏览文件 @
c723f6f4
...
@@ -423,8 +423,8 @@ def gen_random_input(model_output_dir,
...
@@ -423,8 +423,8 @@ def gen_random_input(model_output_dir,
input_file_name
=
"model_input"
):
input_file_name
=
"model_input"
):
for
input_name
in
input_nodes
:
for
input_name
in
input_nodes
:
formatted_name
=
formatted_file_name
(
input_name
,
input_file_name
)
formatted_name
=
formatted_file_name
(
input_name
,
input_file_name
)
if
os
.
path
.
exists
(
formatted_name
):
if
os
.
path
.
exists
(
"%s/%s"
%
(
model_output_dir
,
formatted_name
)
):
sh
.
rm
(
formatted_name
)
sh
.
rm
(
"%s/%s"
%
(
model_output_dir
,
formatted_name
)
)
input_nodes_str
=
","
.
join
(
input_nodes
)
input_nodes_str
=
","
.
join
(
input_nodes
)
input_shapes_str
=
":"
.
join
(
input_shapes
)
input_shapes_str
=
":"
.
join
(
input_shapes
)
generate_input_data
(
"%s/%s"
%
(
model_output_dir
,
input_file_name
),
generate_input_data
(
"%s/%s"
%
(
model_output_dir
,
input_file_name
),
...
@@ -606,8 +606,9 @@ def validate_model(target_soc,
...
@@ -606,8 +606,9 @@ def validate_model(target_soc,
for
output_name
in
output_nodes
:
for
output_name
in
output_nodes
:
formatted_name
=
formatted_file_name
(
formatted_name
=
formatted_file_name
(
output_name
,
output_file_name
)
output_name
,
output_file_name
)
if
os
.
path
.
exists
(
formatted_name
):
if
os
.
path
.
exists
(
"%s/%s"
%
(
model_output_dir
,
sh
.
rm
(
formatted_name
)
formatted_name
)):
sh
.
rm
(
"%s/%s"
%
(
model_output_dir
,
formatted_name
))
adb_pull
(
"%s/%s"
%
(
phone_data_dir
,
formatted_name
),
adb_pull
(
"%s/%s"
%
(
phone_data_dir
,
formatted_name
),
model_output_dir
,
serialno
)
model_output_dir
,
serialno
)
validate
(
platform
,
model_file_path
,
""
,
validate
(
platform
,
model_file_path
,
""
,
...
...
tools/validate.py
浏览文件 @
c723f6f4
...
@@ -75,6 +75,7 @@ def validate_tf_model(platform, mace_runtime, model_file, input_file,
...
@@ -75,6 +75,7 @@ def validate_tf_model(platform, mace_runtime, model_file, input_file,
print
(
"Input graph file '"
+
model_file
+
"' does not exist!"
)
print
(
"Input graph file '"
+
model_file
+
"' does not exist!"
)
sys
.
exit
(
-
1
)
sys
.
exit
(
-
1
)
tf
.
reset_default_graph
()
input_graph_def
=
tf
.
GraphDef
()
input_graph_def
=
tf
.
GraphDef
()
with
open
(
model_file
,
"rb"
)
as
f
:
with
open
(
model_file
,
"rb"
)
as
f
:
data
=
f
.
read
()
data
=
f
.
read
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录