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

paddle模型本来是不固定输入输出的,但是使用Paddle2ONNX转换必须指定input_size,不然会报Due to the operator: pool2d, this model cannot be exported to ONNX错 #1229

Open
wangkd0 opened this issue Apr 19, 2024 · 2 comments
Assignees

Comments

@wangkd0
Copy link

wangkd0 commented Apr 19, 2024

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

问题描述
请在此处详细的描述报错信息
paddle模型本来是不固定输入输出的,但是使用Paddle2ONNX转换必须指定input_size,不然会报Due to the operator: pool2d, this model cannot be exported to ONNX错误

python tools/export.py --config configs/quick_start/ppmattingv2-stdc1-human_512.yml --model_path pretrained_models/ppmattingv2-stdc1-human_512.pdparams --save_dir output/inference_model_end
使用上面这个命令不指定input_size, 导出时就会报错,我不想指定size需要怎么处理

模型使用的ppmattingv2-stdc1-human_512.yml

更多信息 :

  • 用于部署的推理引擎:
  • 为什么需要转换为ONNX格式:
  • 使用onnxruntime运行
  • Paddle2ONNX版本:
  • 你的联系方式(Email/Wechat/Phone):

报错截图

其他信息

@Zheng-Bicheng
Copy link
Collaborator

这个需要固定一下输入shape哈

@Zheng-Bicheng
Copy link
Collaborator

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

No branches or pull requests

2 participants