Ubuntu下编译MNN的前向传播和模型转化工具

前向传播编译:

1. 下载MNN。下载地址:https://github.com/alibaba/MNN/releases(目前最新版本是2.0.0)。

2. 进入MNN文件夹,然后运行下面命令行。

mkdir build && cd build && cmake .. && make -j8

模型转换编译:

cd MNN/
./schema/generate.sh
mkdir build
cd build
cmake .. -DMNN_BUILD_CONVERTER=true && make -j4

模型转换方法:

cd MNN/build 先进入到build文件夹中

TensorFlow -> MNN 命令:
./MNNConvert -f TF --modelFile XXX.pb --MNNModel XXX.mnn --bizCode biz

TensorFlow Lite -> MNN 命令:
./MNNConvert -f TFLITE --modelFile XXX.tflite --MNNModel XXX.mnn --bizCode biz

Caffe -> MNN 命令:
./MNNConvert -f CAFFE --modelFile XXX.caffemodel --prototxt XXX.prototxt --MNNModel XXX.mnn --bizCode biz

ONNX -> MNN 命令:
./MNNConvert -f ONNX --modelFile XXX.onnx --MNNModel XXX.mnn --bizCode biz

TorchScript -> MNN 命令:
./MNNConvert -f TORCH --modelFile XXX.pt --MNNModel XXX.mnn --bizCode biz

Windows下编译MNN的前向传播和模型转化工具

前提已经安装好Visual Studio

前向传播编译:

1. 下载MNN。 下载地址:https://github.com/alibaba/MNN/releases(目前最新版本是2.0.0)。

2. 打开vs的x64命令窗口,如下图所示:

 3. 在命令窗口中,进入到MNN所在的目录。并输入一下命令:

cd MNN/
mkdir build
cd build
cmake -G "NMake Makefiles" -DCMAKE_BUILD_TYPE=Release ..
nmake

模型转换编译:

1. 输入一下命令行即可完成编译:

cd MNN/
cd build
cmake -G "Ninja" -DMNN_BUILD_SHARED_LIBS=OFF -DMNN_BUILD_CONVERTER=ON -DCMAKE_BUILD_TYPE=Release -DMNN_WIN_RUNTIME_MT=ON ..
ninja

模型转换方法:

cd MNN/build 先进入到build文件夹中

TensorFlow -> MNN 命令:
MNNConvert.exe -f TF --modelFile XXX.pb --MNNModel XXX.mnn --bizCode biz

TensorFlow Lite -> MNN 命令:
MNNConvert.exe -f TFLITE --modelFile XXX.tflite --MNNModel XXX.mnn --bizCode biz

Caffe -> MNN 命令:
MNNConvert.exe -f CAFFE --modelFile XXX.caffemodel --prototxt XXX.prototxt --MNNModel XXX.mnn --bizCode biz

ONNX -> MNN 命令:
MNNConvert.exe -f ONNX --modelFile XXX.onnx --MNNModel XXX.mnn --bizCode biz

TorchScript -> MNN 命令:
MNNConvert.exe -f TORCH --modelFile XXX.pt --MNNModel XXX.mnn --bizCode biz

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐