Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
X2Paddle
提交
0680f324
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看板
提交
0680f324
编写于
10月 10, 2022
作者:
W
wjj19950828
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fixed for ci
上级
20eef555
变更
2
展开全部
隐藏空白更改
内联
并排
Showing
2 changed file
with
1621 addition
and
2 deletion
+1621
-2
x2paddle/decoder/onnx_decoder.py
x2paddle/decoder/onnx_decoder.py
+8
-2
x2paddle/decoder/onnx_shape_inference.py
x2paddle/decoder/onnx_shape_inference.py
+1613
-0
未找到文件。
x2paddle/decoder/onnx_decoder.py
浏览文件 @
0680f324
...
@@ -13,6 +13,7 @@
...
@@ -13,6 +13,7 @@
# limitations under the License.
# limitations under the License.
from
x2paddle.core.graph
import
GraphNode
,
Graph
from
x2paddle.core.graph
import
GraphNode
,
Graph
from
x2paddle.decoder.onnx_shape_inference
import
SymbolicShapeInference
from
onnx.checker
import
ValidationError
from
onnx.checker
import
ValidationError
from
onnx.checker
import
check_model
from
onnx.checker
import
check_model
from
onnx
import
helper
,
shape_inference
from
onnx
import
helper
,
shape_inference
...
@@ -184,8 +185,13 @@ class ONNXGraph(Graph):
...
@@ -184,8 +185,13 @@ class ONNXGraph(Graph):
self
.
graph
=
onnx_model
.
graph
self
.
graph
=
onnx_model
.
graph
self
.
get_place_holder_nodes
()
self
.
get_place_holder_nodes
()
print
(
"Shape inferencing ..."
)
print
(
"Shape inferencing ..."
)
onnx_model
=
shape_inference
.
infer_shapes
(
onnx_model
)
try
:
self
.
graph
=
onnx_model
.
graph
self
.
graph
=
SymbolicShapeInference
.
infer_shapes
(
onnx_model
,
fixed_input_shape
=
self
.
fixed_input_shape
)
except
:
print
(
'[WARNING] Shape inference by ONNX offical interface.'
)
onnx_model
=
shape_inference
.
infer_shapes
(
onnx_model
)
self
.
graph
=
onnx_model
.
graph
print
(
"Shape inferenced."
)
print
(
"Shape inferenced."
)
self
.
build
()
self
.
build
()
self
.
collect_value_infos
()
self
.
collect_value_infos
()
...
...
x2paddle/decoder/onnx_shape_inference.py
0 → 100644
浏览文件 @
0680f324
此差异已折叠。
点击以展开。
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录