查资料发现貌似pnnx转ncnn比较方便一些,算子实现更全,避免了onnx中间转换,所以就去找pnnx工具。
一开始是想找一个直接能用的工具,就去下载了编译好的:
Release Release 20220418 · pnnx/pnnx · GitHub
解压后使用发现,python版本貌似对不上,运行 ./pnnx报错
dyld: Library not loaded: /usr/local/opt/python@3.9/Frameworks/Python.framework/Versions/3.9/Python
其实这个问题应该能通过替换目录解决,但不想折腾本地的环境了,就想自己直接编译吧。
找了一圈发现了nihui大佬的教程,跟着做了一遍
Windows/Linux/MacOS 编译 PNNX 步骤 - 知乎
编译是成功了,但把libtorch的库粘到pnnx运行程序同目录下,发现链接不到动态库,我对cmake不太了解,我觉得应该是能改一下CMakeLists.txt,指定一下动态链接库来解决的,后来又找到一个编译方法,就想先试试新的,如果不行再回来改cmake,不过最后一个方法成功了,参考了这个大佬的链接:
这个方法是自己去找python的,我本地anaconda装了好多虚拟环境,但发现它自己找到的python就是本地base的python,我也不知道改哪才能让它找别的虚拟环境的python,最后还是放弃了,在base环境pip install torch cmake ninja
之前编译ncnn的时候就下载了代码 (https://github.com/Tencent/ncnn.git)
直接去文件里跟着做了一遍,非常简单,就编译好了。
总结一下,觉得上面的三个方法应该都是可行的,但我的工程能力实在是有点太可怜了,有空的话再试试看,研究下人家的CMakeLists.txt是怎么写的