Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleClas
提交
68a4c7fe
P
PaddleClas
项目概览
PaddlePaddle
/
PaddleClas
大约 1 年 前同步成功
通知
115
Star
4999
Fork
1114
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
19
列表
看板
标记
里程碑
合并请求
6
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleClas
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
19
Issue
19
列表
看板
标记
里程碑
合并请求
6
合并请求
6
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
68a4c7fe
编写于
6月 17, 2022
作者:
C
cuicheng01
提交者:
GitHub
6月 17, 2022
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #1976 from HydrogenSulfate/polish_onnx_doc
polish paddle2onnx doc
上级
95edc0d6
6d4bfeae
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
33 addition
and
27 deletion
+33
-27
deploy/paddle2onnx/readme.md
deploy/paddle2onnx/readme.md
+33
-27
未找到文件。
deploy/paddle2onnx/readme.md
浏览文件 @
68a4c7fe
# paddle2onnx 模型转化与预测
本章节介绍 ResNet50_vd 模型如何转化为 ONNX 模型,并基于 ONNX 引擎预测。
## 目录
-
[
paddle2onnx 模型转化与预测
](
#paddle2onnx-模型转化与预测
)
-
[
1. 环境准备
](
#1-环境准备
)
-
[
2. 模型转换
](
#2-模型转换
)
-
[
3. onnx 预测
](
#3-onnx-预测
)
## 1. 环境准备
需要准备 Paddle2ONNX 模型转化环境,和 ONNX 模型预测环境。
Paddle2ONNX 支持将 PaddlePaddle
模型格式转化到 ONNX 模型格式,算子目前稳定支持导出 ONNX Opset 9~11,部分Paddle算子支持更低的ONNX Opset转换
。
更多细节可参考
[
Paddle2ONNX
](
https://github.com/PaddlePaddle/Paddle2ONNX
/blob/develop/README_zh.md
)
Paddle2ONNX 支持将 PaddlePaddle
inference 模型格式转化到 ONNX 模型格式,算子目前稳定支持导出 ONNX Opset 9~11
。
更多细节可参考
[
Paddle2ONNX
](
https://github.com/PaddlePaddle/Paddle2ONNX
#paddle2onnx
)
-
安装 Paddle2ONNX
```
python3.7 -m pip install paddle2onnx
```
```
shell
python3.7
-m
pip
install
paddle2onnx
```
-
安装 ONNX 运行时
```
python3.7 -m pip install onnxruntime
```
-
安装 ONNX 推理引擎
```
shell
python3.7
-m
pip
install
onnxruntime
```
下面以 ResNet50_vd 为例,介绍如何将 PaddlePaddle inference 模型转换为 ONNX 模型,并基于 ONNX 引擎预测。
## 2. 模型转换
-
ResNet50_vd inference模型下载
```
cd deploy
mkdir models && cd models
wget -nc https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/inference/ResNet50_vd_infer.tar && tar xf ResNet50_vd_infer.tar
cd ..
```
```shell
cd deploy
mkdir models && cd models
wget -nc https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/inference/ResNet50_vd_infer.tar && tar xf ResNet50_vd_infer.tar
cd ..
```
-
模型转换
使用 Paddle2ONNX 将 Paddle 静态图模型转换为 ONNX 模型格式:
```
paddle2onnx --model_dir=./models/ResNet50_vd_infer/ \
--model_filename=inference.pdmodel \
--params_filename=inference.pdiparams \
--save_file=./models/ResNet50_vd_infer/inference.onnx \
--opset_version=10 \
--enable_onnx_checker=True
```
使用 Paddle2ONNX 将 Paddle 静态图模型转换为 ONNX 模型格式:
```shell
paddle2onnx --model_dir=./models/ResNet50_vd_infer/ \
--model_filename=inference.pdmodel \
--params_filename=inference.pdiparams \
--save_file=./models/ResNet50_vd_infer/inference.onnx \
--opset_version=10 \
--enable_onnx_checker=True
```
执行完毕后,
ONNX 模型
`inference.onnx`
会被保存在
`./models/ResNet50_vd_infer/`
路径下
转换完毕后,生成的
ONNX 模型
`inference.onnx`
会被保存在
`./models/ResNet50_vd_infer/`
路径下
## 3. onnx 预测
执行如下命令:
```
```
shell
python3.7 python/predict_cls.py
\
-c
configs/inference_cls.yaml
\
-o
Global.use_onnx
=
True
\
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录