Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Xiaomi
Mace
提交
6479f2e3
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,发现更多精彩内容 >>
提交
6479f2e3
编写于
3月 07, 2018
作者:
L
liuqi
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Use yaml basic array format for input_node, etc.
上级
ce464eec
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
14 addition
and
5 deletion
+14
-5
example.yaml
example.yaml
+8
-4
mace_tools.py
mace_tools.py
+6
-1
未找到文件。
example.yaml
浏览文件 @
6479f2e3
...
...
@@ -25,10 +25,14 @@ models:
weight_file_path
:
path/to/weight.caffemodel
model_sha256_checksum
:
05d92625809dc9edd6484882335c48c043397aed450a168d75eb8b538e86881a
weight_sha256_checksum
:
05d92625809dc9edd6484882335c48c043397aed450a168d75eb8b538e86881a
input_node
:
input_node0,input_node1
output_node
:
output_node0,output_node1
input_shape
:
1,256,256,3:1,128,128,3
output_shape
:
1,256,256,2:1,1,1,2
input_node
:
[
input_node0
,
input_node1
]
output_node
:
[
output_node0
,
output_node1
]
input_shape
:
-
1,256,256,3
-
1,128,128,3
output_shape
:
-
1,256,256,2
-
1,1,1,2
runtime
:
cpu
limit_opencl_kernel_time
:
1
dsp_mode
:
0
...
...
mace_tools.py
浏览文件 @
6479f2e3
...
...
@@ -205,6 +205,11 @@ def main(unused_args):
os
.
environ
[
"MODEL_TAG"
]
=
model_name
model_config
=
configs
[
"models"
][
model_name
]
for
key
in
model_config
:
if
key
in
[
'input_node'
,
'output_node'
]
and
isinstance
(
model_config
[
key
],
list
):
os
.
environ
[
key
.
upper
()]
=
","
.
join
(
model_config
[
key
])
elif
key
in
[
'input_shape'
,
'output_shape'
]
and
isinstance
(
model_config
[
key
],
list
):
os
.
environ
[
key
.
upper
()]
=
":"
.
join
(
model_config
[
key
])
else
:
os
.
environ
[
key
.
upper
()]
=
str
(
model_config
[
key
])
md5
=
hashlib
.
md5
()
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录