MNN框架在windows上的部署

笑眯眯办大事發表於2024-08-16
框架Windows

一、MNN.lib檔案等的生成

算是第一次接觸MNN框架部署這部分內容吧，對整個流程都不是很理解，故先從官方文件入手
先從github下載官方的專案，地址為：

https://github.com/alibaba/MNN

使用visual studio的powershell開啟，執行官網給出的指令

cd path/to/MNN
powershell ./schema/generate.ps1
mkdir build
cd build
cmake -G "NMake Makefiles" -DCMAKE_BUILD_TYPE=Release -DMNN_BUILD_DEMO=ON ..
nmake

這會在工程目錄下生成一個build檔案，裡面會建立一系列的內容，比較重要的有：
MNN.lib等

之後就可以進行demo程式的執行了。

這裡簡單瞭解一下使用MNN進行機器學習的流程，為：

  • 訓練
  • 轉換
  • 推理
    細節內容介紹檢視官方文件：
https://www.yuque.com/mnn/cn/usage

這裡執行demo只是執行了轉換和推理的流程，訓練最好不要使用MNN框架，而是選擇較為成熟的TensorFlow、PyTorch等。

二、模型轉換

模型轉換需要用到“ProtoBuf”，參考連結：

https://blog.csdn.net/weixin_74531333/article/details/140469169

之後參考官方readme

cd MNN
mkdir build
cd build
cmake .. -DMNN_BUILD_CONVERTER=true
make

這裡使用cmake和make指令需要重點注意，由於win上面本身不支援make指令，所以需要使用MinGw工具來提供make指令的支援，參考連結：

https://blog.csdn.net/linjf520/article/details/108559210

https://blog.csdn.net/heiwa110/article/details/130696242

並且camke指令也需要修改，CMake生成的Makefile可能預設是為NMake或其他Windows工具鏈生成的，而不是針對MinGW的make工具
所以將指令修改為：

cmake -G "MinGW Makefiles" -DMNN_BUILD_CONVERTER=true ..

相關文章