Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
孤小逸
Mace
提交
217b3fd0
Mace
项目概览
孤小逸
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
1
Star
1
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,发现更多精彩内容 >>
提交
217b3fd0
编写于
7月 03, 2018
作者:
L
liuqi
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Fix bug: need tensorflow when convert caffe model.
上级
82bf1768
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
20 addition
and
24 deletion
+20
-24
mace/python/tools/convert_util.py
mace/python/tools/convert_util.py
+0
-20
mace/python/tools/converter.py
mace/python/tools/converter.py
+3
-3
mace/python/tools/tf_dsp_converter_lib.py
mace/python/tools/tf_dsp_converter_lib.py
+17
-1
未找到文件。
mace/python/tools/convert_util.py
浏览文件 @
217b3fd0
...
...
@@ -13,26 +13,6 @@
# limitations under the License.
import
tensorflow
as
tf
from
mace.proto
import
mace_pb2
TF_DTYPE_2_MACE_DTYPE_MAP
=
{
tf
.
float32
:
mace_pb2
.
DT_FLOAT
,
tf
.
half
:
mace_pb2
.
DT_HALF
,
tf
.
int32
:
mace_pb2
.
DT_INT32
,
tf
.
qint32
:
mace_pb2
.
DT_INT32
,
tf
.
quint8
:
mace_pb2
.
DT_UINT8
,
tf
.
uint8
:
mace_pb2
.
DT_UINT8
,
}
def
tf_dtype_2_mace_dtype
(
tf_dtype
):
mace_dtype
=
TF_DTYPE_2_MACE_DTYPE_MAP
.
get
(
tf_dtype
,
None
)
if
not
mace_dtype
:
raise
Exception
(
"Not supported tensorflow dtype: "
+
tf_dtype
)
return
mace_dtype
def
mace_check
(
condition
,
msg
):
if
not
condition
:
raise
Exception
(
msg
)
mace/python/tools/converter.py
浏览文件 @
217b3fd0
...
...
@@ -19,12 +19,9 @@ import os.path
import
copy
from
mace.proto
import
mace_pb2
from
mace.python.tools
import
tf_dsp_converter_lib
from
mace.python.tools
import
memory_optimizer
from
mace.python.tools
import
model_saver
from
mace.python.tools.converter_tool
import
base_converter
as
cvt
from
mace.python.tools.converter_tool
import
tensorflow_converter
from
mace.python.tools.converter_tool
import
caffe_converter
from
mace.python.tools.converter_tool
import
transformer
from
mace.python.tools.convert_util
import
mace_check
...
...
@@ -101,6 +98,7 @@ def main(unused_args):
if
FLAGS
.
runtime
==
'dsp'
:
if
FLAGS
.
platform
==
'tensorflow'
:
from
mace.python.tools
import
tf_dsp_converter_lib
output_graph_def
=
tf_dsp_converter_lib
.
convert_to_mace_pb
(
FLAGS
.
model_file
,
FLAGS
.
input_node
,
FLAGS
.
output_node
,
FLAGS
.
dsp_mode
)
...
...
@@ -132,9 +130,11 @@ def main(unused_args):
option
.
add_output_node
(
output_node
)
if
FLAGS
.
platform
==
'tensorflow'
:
from
mace.python.tools.converter_tool
import
tensorflow_converter
converter
=
tensorflow_converter
.
TensorflowConverter
(
option
,
FLAGS
.
model_file
)
elif
FLAGS
.
platform
==
'caffe'
:
from
mace.python.tools.converter_tool
import
caffe_converter
converter
=
caffe_converter
.
CaffeConverter
(
option
,
FLAGS
.
model_file
,
FLAGS
.
weight_file
)
...
...
mace/python/tools/tf_dsp_converter_lib.py
浏览文件 @
217b3fd0
...
...
@@ -19,13 +19,29 @@ from tensorflow import gfile
from
operator
import
mul
from
dsp_ops
import
DspOps
from
mace.python.tools
import
graph_util
from
mace.python.tools.convert_util
import
tf_dtype_2_mace_dtype
# converter --input ../libcv/quantized_model.pb \
# --output quantized_model_dsp.pb \
# --runtime dsp --input_node input_node \
# --output_node output_node
TF_DTYPE_2_MACE_DTYPE_MAP
=
{
tf
.
float32
:
mace_pb2
.
DT_FLOAT
,
tf
.
half
:
mace_pb2
.
DT_HALF
,
tf
.
int32
:
mace_pb2
.
DT_INT32
,
tf
.
qint32
:
mace_pb2
.
DT_INT32
,
tf
.
quint8
:
mace_pb2
.
DT_UINT8
,
tf
.
uint8
:
mace_pb2
.
DT_UINT8
,
}
def
tf_dtype_2_mace_dtype
(
tf_dtype
):
mace_dtype
=
TF_DTYPE_2_MACE_DTYPE_MAP
.
get
(
tf_dtype
,
None
)
if
not
mace_dtype
:
raise
Exception
(
"Not supported tensorflow dtype: "
+
tf_dtype
)
return
mace_dtype
padding_mode
=
{
'NA'
:
0
,
'SAME'
:
1
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录