Python的示例
从PyTorch GitHub官网下载好mobilenet.py模型, pytorch 官网vision/torchvision/models/中的mobilenet.py文件。
python mobilenet_test.py
python model2onnx.py
这是在conda的python环境中安装MNN。除了python能import调用API执行模型计算之外,还直接能在命令行操作MNN的工具。
pip install MNN
使用 mnnconvert -h 查看一些参数:
转换onnx模型: mnnconvert -f ONNX --modelFile mobilenet_v2-b0353104.onnx --MNNModel mobilenet_v2-b0353104.mnn --bizCode MNN
:
好玩的是执行python mnn_test.py
遇到了错误,查阅资料发现是输入需要是NC4HW4的格式,改一下即可。
pip中安装的onnx有时候转换模型并不是很好用,还是推荐从源码编译安装MNN。