search

DefTruth / lite.ai.toolkit

🛠 A lite C++ toolkit of awesome AI models, support ONNXRuntime, MNN, TNN, NCNN and TensorRT.
https://github.com/DefTruth/lite.ai.toolkit
GNU General Public License v3.0
3.64k stars 695 forks source link

模型转换问题 #303

Closed ChaoII closed 2 years ago

ChaoII commented 2 years ago

请问一下，很多模型从onnx转ncnn都存在很多困难，请问作者是如何转换的啊？onnx-sim也不好使

DefTruth commented 2 years ago

是有这个问题，这个可能和ncnn的设计有关，ncnn::Mat假设batch=1，然后只有h,w,c，三个维度，所以最多只能处理(1,h,w,c)这样的张量。如果你的模型中存在了很多超过4维的张量，或者有很多slices操作.... 是需要手动先去除这些操作才能转换成功的。

DefTruth commented 2 years ago

另外，建议可以尝试一下使用nihui大佬的pnnx来做模型转换啊：