代码拉取完成,页面将自动刷新
同步操作将从 PaddlePaddle/PaddleSeg 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
简体中文 | English
PaddleSeg训练好模型后,也支持导出ONNX格式模型,本教程提供一个示例介绍使用方法。
导出ONNX格式模型的完整方法,请参考Paddle2ONNX。
参考文档导出预测模型。
复用文档中的示例,成功将导出的预测模型文件保存在output文件夹中,如下。
./output
├── deploy.yaml # 部署相关的配置文件,主要说明数据预处理的方式
├── model.pdmodel # 预测模型的拓扑结构文件
├── model.pdiparams # 预测模型的权重文件
└── model.pdiparams.info # 参数额外信息,一般无需关注
安装Paddle2ONNX(高于或等于0.6版本)。
pip install paddle2onnx
执行如下命令,使用Paddle2ONNX将output文件夹中的预测模型导出为ONNX格式模型。
paddle2onnx --model_dir output \
--model_filename model.pdmodel \
--params_filename model.pdiparams \
--opset_version 11 \
--save_file output.onnx
导出的ONNX格式模型保存为output.onnx文件。
参考文档:
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。