Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
48e0e5bf
X
X2Paddle
项目概览
PaddlePaddle
/
X2Paddle
1 年多 前同步成功
通知
328
Star
698
Fork
167
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
26
列表
看板
标记
里程碑
合并请求
4
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
X
X2Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
26
Issue
26
列表
看板
标记
里程碑
合并请求
4
合并请求
4
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
48e0e5bf
编写于
4月 03, 2019
作者:
J
Jason
提交者:
GitHub
4月 03, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update README.md
上级
0c9e1572
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
10 addition
and
6 deletion
+10
-6
tensorflow2fluid/README.md
tensorflow2fluid/README.md
+10
-6
未找到文件。
tensorflow2fluid/README.md
浏览文件 @
48e0e5bf
...
...
@@ -4,7 +4,7 @@
tensorflow2fluid支持将训练好的TensorFlow模型转换为PaddlePaddle模型,包括基于PaddlePaddle实现的模型前向计算网络python代码,以及PaddlePaddle可加载的模型参数文件。
此外在
[
[doc
](
doc/接口速查表.md
)
]目录中整理了TensorFlow-PaddlePaddle的常用API对比分析。
[
环境安装
](
#环境安装
)
[使用方法](#使用方法)
[验证模型](#验证模型)
[
注意事项
](
#注意事项
)
[
环境安装
](
#环境安装
)
[使用方法](#使用方法)
[验证模型](#验证模型)
[
常见问题
](
#常见问题
)
## 环境安装
...
...
@@ -107,22 +107,26 @@ tensorflow2fluid在如下tensorflow模型上测试了模型转换前后的diff
| | YOLO-V3 |
[
code
](
https://github.com/mystic123/tensorflow-yolo-v3
)
| 6.20E-04 |
| 语义分割 | Unet |
[
code
](
https://github.com/jakeret/tf_unet
)
| 4.17E-07 |
##
注意事项
1.
转换参数
`input_format`
的设定
##
常见问题
1.
转换参数
`input_format`
的设定
?
> TensorFlow中的CV模型,大多采用`NHWC`的输入格式,但同时也可以支持`NCHW`的格式输入;而在PaddlePaddle中,支持的是`NCHW`的格式。因此需要在转换模型时,指定TensorFlow模型的输入格式,转换过程中会根据输入格式,对输入数据,参数进行变换。
2.
转换参数
`input_shape`
的设定
2.
转换参数
`input_shape`
的设定
?
> 在模型转换时,需设定输入数据的具体`shape`。因为转换过程中,涉及到较多参数的转换,因此模型转换完成应用到预测时,输入数据的`shape`也须与之前指定的一致,否则可能会出错。
3.
转换参数
`use_cuda`
的设定
3.
转换参数
`use_cuda`
的设定
?
> 受限于PaddlePaddle与TensorFlow部分OP上的实现差异,部分tensor参数(在TensorFlow中,这部分参数类型是tensor类型,但值保持不变)需要通过infer得到。因此模型转换过程中,同时也会加载tensorflow模型进行预测,消耗计算资源。在有GPU资源的的前提下,将`use_cuda`设为`True`有助于提升转换速度。
4.
模型转换前后diff对比
4.
模型转换前后diff对比
?
> tensorflow2fluid仍在不断完善和测试中,用户转换完模型后,注意对比模型在转换前后的输出diff是否在可接受范围内。此外转换后的模型结构`mymodel.py`如存在构建失败的问题,可能是由于部分参数在特殊情况下未被考虑到导致,用户可以直接通过修改`mymodel.py`来解决。
5.
模型转换失败,提示"Unsupported OP: XXX"?
> 目前tf2fluid支持50个左右常见OP的转换,仍然在不断补充中,当出现如上提示时,即表示模型存在暂未支持的OP,用户可以直接在[tf2fluid/paddle_emitter.py]中仿照`emit_xxx`函数添加转换代码支持,或者也欢迎通过提ISSUE的方式让我们知道你的需求!
## Link
本目录下部分代码参考了MMdnn-Tensorflow,对此表示感谢!
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录