Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

PP-MattingV2模型重训练后用onnxruntime进行推理失败 #1498

Open
LiaoYuWei1 opened this issue Feb 10, 2025 · 1 comment
Open

PP-MattingV2模型重训练后用onnxruntime进行推理失败 #1498

LiaoYuWei1 opened this issue Feb 10, 2025 · 1 comment

Comments

@LiaoYuWei1
Copy link

请将下面信息填写完整,便于我们快速解决问题,谢谢!

问题描述
你好,我在对pp-MattingV2模型的配置文件进行修改后对模型进行了重训练,将预处理的resize从512限制修改为256限制,目前遇到在paddle框架下能够正常推理,但是用paddle2onnx转成onnx后,用onnxruntime进行推理时,模型推理失败且程序崩溃退出,目前暂时定位到是在dpp模块出现的问题,但是具体的问题原因暂时没有排查到。能否帮忙排查是模型问题还是paddle2onnx的工具bug

更多信息 :

  • 用于部署的推理引擎:
  • 为什么需要转换为ONNX格式:尝试在本地进行c++推理部署,测试效率
  • Paddle2ONNX版本:V1.3.1
  • 你的联系方式(Email/Wechat/Phone):[email protected]

报错截图

Image

其他信息
同样的环境下,预训练模型转换成onnx并使用onnxruntime进行onnx推理验证并未出现类似的问题

@risemeup1
Copy link
Collaborator

等三月10号发版本后,测试一下能不能过,目前在更新

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants