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

转ncnn不成功 #5747

Open
qiuzhewei opened this issue Oct 18, 2024 · 0 comments
Open

转ncnn不成功 #5747

qiuzhewei opened this issue Oct 18, 2024 · 0 comments

Comments

@qiuzhewei
Copy link

error log | 日志或报错信息 | ログ

你好,
我是使用paddle训练得到的模型,然后转成onnx模型,onnx模型和paddle模型的一致性是对的上的。
然后我用命令pnnx test_model/picodet.onnx inputshape=[1,3,320,320]f32[1,2]f32 fp16=0 能够正常生成所有文件,但是我运行picodet_ncnn.py时报错
layer pnnx.Expression not exists or registered

运行picodet_ncnn.py时报错
Traceback (most recent call last): File "test_model/picodet_pnnx.py", line 648, in <module> print(test_inference()) File "test_model/picodet_pnnx.py", line 645, in test_inference return net(v_0, v_1) File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(*args, **kwargs) File "test_model/picodet_pnnx.py", line 578, in forward v_247, v_248 = torch.split(tensor=v_246, dim=2, split_size_or_sections=None) File "/usr/local/lib/python3.8/dist-packages/torch/functional.py", line 189, in split return tensor.split(split_size_or_sections, dim) File "/usr/local/lib/python3.8/dist-packages/torch/_tensor.py", line 803, in split return torch._VF.split_with_sizes(self, split_size, dim) TypeError: split_with_sizes(): argument 'split_sizes' (position 2) must be tuple of ints, not NoneType
有什么解决办法嘛?

model | 模型 | モデル

  1. original model
    picodet_s

how to reproduce | 复现步骤 | 再現方法

  1. 去paddledetection官网下载对应的模型
    2.转成onnx模型
    3.再使用pnnx把onnx模型转成ncnn模型
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

1 participant