Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
eb2eb0b8
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看板
提交
eb2eb0b8
编写于
12月 30, 2021
作者:
W
wjj19950828
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fixed for CI
上级
f81f905b
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
7 addition
and
6 deletion
+7
-6
docs/inference_model_convertor/demo/tensorflow2paddle.ipynb
docs/inference_model_convertor/demo/tensorflow2paddle.ipynb
+2
-2
x2paddle/core/program.py
x2paddle/core/program.py
+3
-3
x2paddle/decoder/onnx_decoder.py
x2paddle/decoder/onnx_decoder.py
+2
-1
未找到文件。
docs/inference_model_convertor/demo/tensorflow2paddle.ipynb
浏览文件 @
eb2eb0b8
...
@@ -81,7 +81,7 @@
...
@@ -81,7 +81,7 @@
"source": [
"source": [
"## 模型迁移\n",
"## 模型迁移\n",
"### 1. 获取MobileNetV1的FrozenModel\n",
"### 1. 获取MobileNetV1的FrozenModel\n",
"由于X2Paddle只支持TensorFlow中FrozenModel的转换,如果为纯checkpoint模型,需要参考参考X2Paddle官方[文档](https://github.com/PaddlePaddle/X2Paddle/blob/
develop
/docs/user_guides/export_tf_model.md),将其转换为FrozenModel,本示例中提供的模型为FrozenModel,所以无需转换。"
"由于X2Paddle只支持TensorFlow中FrozenModel的转换,如果为纯checkpoint模型,需要参考参考X2Paddle官方[文档](https://github.com/PaddlePaddle/X2Paddle/blob/
release-1.1
/docs/user_guides/export_tf_model.md),将其转换为FrozenModel,本示例中提供的模型为FrozenModel,所以无需转换。"
]
]
},
},
{
{
...
@@ -210,4 +210,4 @@
...
@@ -210,4 +210,4 @@
},
},
"nbformat": 4,
"nbformat": 4,
"nbformat_minor": 4
"nbformat_minor": 4
}
}
\ No newline at end of file
x2paddle/core/program.py
浏览文件 @
eb2eb0b8
...
@@ -532,9 +532,9 @@ class PaddleGraph(object):
...
@@ -532,9 +532,9 @@ class PaddleGraph(object):
paddle
.
save
(
self
.
parameters
,
save_path
)
paddle
.
save
(
self
.
parameters
,
save_path
)
def
dygraph2static
(
self
,
save_dir
,
input_shapes
=
[],
input_types
=
[]):
def
dygraph2static
(
self
,
save_dir
,
input_shapes
=
[],
input_types
=
[]):
s
ep
c_list
=
list
()
s
pe
c_list
=
list
()
for
i
,
name
in
enumerate
(
self
.
inputs
):
for
i
,
name
in
enumerate
(
self
.
inputs
):
s
ep
c_list
.
append
(
s
pe
c_list
.
append
(
paddle
.
static
.
InputSpec
(
paddle
.
static
.
InputSpec
(
shape
=
input_shapes
[
i
],
name
=
name
,
dtype
=
input_types
[
i
]))
shape
=
input_shapes
[
i
],
name
=
name
,
dtype
=
input_types
[
i
]))
path
=
osp
.
abspath
(
save_dir
)
path
=
osp
.
abspath
(
save_dir
)
...
@@ -548,7 +548,7 @@ class PaddleGraph(object):
...
@@ -548,7 +548,7 @@ class PaddleGraph(object):
else
:
else
:
model
.
set_dict
(
restore
)
model
.
set_dict
(
restore
)
model
.
eval
()
model
.
eval
()
static_model
=
paddle
.
jit
.
to_static
(
model
,
input_spec
=
s
ep
c_list
)
static_model
=
paddle
.
jit
.
to_static
(
model
,
input_spec
=
s
pe
c_list
)
try
:
try
:
paddle
.
jit
.
save
(
static_model
,
paddle
.
jit
.
save
(
static_model
,
osp
.
join
(
save_dir
,
"inference_model/model"
))
osp
.
join
(
save_dir
,
"inference_model/model"
))
...
...
x2paddle/decoder/onnx_decoder.py
浏览文件 @
eb2eb0b8
...
@@ -583,7 +583,8 @@ class ONNXDecoder(object):
...
@@ -583,7 +583,8 @@ class ONNXDecoder(object):
item
.
name
=
self
.
make_variable_name
(
item
.
name
)
item
.
name
=
self
.
make_variable_name
(
item
.
name
)
for
node
in
graph
.
node
:
for
node
in
graph
.
node
:
node
.
name
=
node
.
output
[
0
]
node
.
name
=
node
.
output
[
0
]
if
":"
in
node
.
name
and
len
(
node
.
output
)
>
1
:
if
":"
in
node
.
name
and
len
(
node
.
output
)
>
1
and
node
.
op_type
!=
"LSTM"
:
node
.
name
=
node
.
name
.
split
(
':'
)[
0
]
node
.
name
=
node
.
name
.
split
(
':'
)[
0
]
node
.
name
=
self
.
make_variable_name
(
node
.
name
)
node
.
name
=
self
.
make_variable_name
(
node
.
name
)
for
i
in
range
(
len
(
node
.
input
)):
for
i
in
range
(
len
(
node
.
input
)):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录