- 依赖项
- cmake 3.0+
- protobuf (protoc 3.0+)
- gcc/g++ 4.9+
- 下载源码
git clone https://github.com/alibaba/MNN.git - 编译MNN模型转化器:MNNConvert,整个编译过程需要几分钟
cd MNN
./schema/generate.sh
mkdir build
cd build
cmake .. -DMNN_BUILD_CONVERTER= true&& make -j4
- 模型转换
# TensorFlow -> MNN
./MNNConvert -f TF --modelFile path/to/XXX.pb --MNNModel path/to/XXX.mnn --bizCode MNN
# TensorFlow Lite -> MNN
./MNNConvert -f TFLITE --modelFile XXX.tflite --MNNModel XXX.mnn --bizCode MNN
# Caffe -> MNN
./MNNConvert -f CAFFE --modelFile XXX.caffemodel --prototxt XXX.prototxt --MNNModel XXX.mnn --bizCode MNN
# ONNX -> MNN
./MNNConvert -f ONNX --modelFile XXX.onnx --MNNModel XXX.mnn --bizCode MNN
- 测试MNNDump2Json
MNNDump2Json用于将MNN二进制模型文件转储为可读的json格式。 与原始模型参数进行比较有助于检查模型转换是否正确
./MNNDump2Json path/to/xxx.mnn path/to/xxx.json