mnn 编译转换工具

 

这个使用cmake编译的,问题是不能编译Release版本

1.在github上下载protobuf,得到protobuf-master文件夹,在此文件夹下新建一个vsproject文件夹(文件夹名你随便取,只是因为此文件夹下已经有了一个build的东东,所以才起的这个名字),然后按照上篇博客里的描述的那样使用cmake-gui.exe对protobuf生成vs的工程。source code选择protobuf目录下的cmake文件夹,其余的操作都如同上一篇的那样。设置如下:


然后在vs中对刚才生成的工程进行编译,记得选release哦。最后得到如下文件目录:

mnn 编译转换工具_第1张图片

直到这一步,基本上不会出现任何问题,一路绿灯

将上一步得到的“include”和“Release”文件夹路径写入你的环境变量,一定要做!!!编译MNN的转换工具需要!最后我的环境路径如下:


去MNN官网下载mnn的包吧,解压后得到MNN-master文件夹

默认你已经安装好了python3和vs2017开始编译
在MNN-master文件夹下定位到“schema”下,新建一个python文件,文件名:generate.py 内容为:

#-*-coding:utf-8-*-
#coding by: yuangu([email protected])

import os
import sys
import shutil
import platform

def p():
    frozen = "not"
    if getattr(sys, 'frozen',False):
        frozen = "ever so"
        return os.path.dirname(sys.executable)

    return os.path.split(os.path.realpath(__file__))[0]

currentWorkPath = p()
os.chdir(currentWorkPath)

if '-lazy' in sys.argv and os.path.isdir("current"):
    print("*** done ***")
    exit(0)

# check is flatbuffer installed or not
FLATC = '../3rd_party/flatbuffers/tmp/flatc' + ('.exe' if "Windows" ==  platform.system() else '')
FLATC = os.path.realpath(FLATC)

if not os.path.isfile(FLATC):
    print("*** building flatc ***")
    tmpDir = os.path.realpath('../3rd_party/flatbuffers/tmp')
    
    if os.path.isdir(tmpDir):
        shutil.rmtree(tmpDir)
    
    os.mkdir(tmpDir)
    os.chdir(tmpDir)

    os.system('cmake  -DCMAKE_BUILD_TYPE=Release ..')
    if "Windows" ==  platform.system():
        os.system('cmake --build . --target flatc --config Release')
        if os.path.isfile( os.path.join(tmpDir, 'Release/flatc.exe') ):
            shutil.move(os.path.join(tmpDir, 'Release/flatc.exe'), FLATC)
    else:
        os.system('cmake --build . --target flatc')


    # dir recover
    os.chdir(currentWorkPath)

# determine directory to use
DIR='default'
if os.path.isdir('private'):
    DIR = 'private'
DIR = os.path.realpath(DIR)

# clean up
print('*** cleaning up ***')
if os.path.isdir('current'):
    shutil.rmtree('current')
os.mkdir('current')

# flatc all fbs
os.chdir('current')
listFile = os.listdir(DIR)
for fileName in listFile:
    tmpFileName = os.path.join(DIR, fileName)
    cmd = "%s -c -b --gen-object-api --reflect-names %s" %(FLATC, tmpFileName)
    os.system(cmd)

os.chdir(currentWorkPath)
print( "*** done ***")





保存后,打开cmd,运行 python generate.py,会自动帮你生成后续编译需要的工程,一般这里也不会有什么错误。这一步想了解具体过程的请参加:https://uzshare.com/view/810473 我是直接拷贝过来用的

进入主题
在命令行下定位到mnn的主目录

输入:
mkdir build && cd build
cmake -G "Visual Studio 15 Win64" -DCMAKE_BUILD_TYPE=Release -DMNN_BUILD_CONVERTER=true -DMNN_BUILD_SHARED_LIBS=false ..

解释:
-G “Visual Studio 15 Win64” 使用64位编译器
-DCMAKE_BUILD_TYPE=Release 编译成release版本
-DMNN_BUILD_CONVERTER=true 编译converter工具
-DMNN_BUILD_SHARED_LIBS 官网上给出的,没细究
命令最后有两个点别忘了!!
等待完成。
完成后就输入:

cmake --build .

这里官网给出的是使用ninja命令编译,这个没用过也没时间去深究了,就直接cmake吧。
如果一切正常的话在build目录下会生成一个Release文件夹,里面就是编译出来的各种工具了。但是在执行cmake --build的时候很可能会报各种奇怪的错误,一般情况下除了那个MNNConvert相关的工程外都能正常编译,主要是因为这个转换的工程依赖于第一步的那个protobuf,列举几个经常碰见的问题:

找不到头文件,检查一下是否把第一步编译出来的Release和include文件夹路径写入了环境路径并且已经生效
报:libprotobuf.lib(any.obj) : error LNK2038: 检测到“RuntimeLibrary”的不匹配项,仔细看一下错误的原因,这里很可能是使用libprotobuf的debug版本来编译MNNConvert的Release版本,反之亦然。反正就是版本不对。我就是碰到了这个问题,可是我明明已经指定了DCMAKE_BUILD_TYPE=Release选项,不得已,在vs2017中打开的工程,选择release编译通过。
报了很多什么参数不匹配,XX函数未定义这一类的错误,建议还是把build文件夹全部删除,重头来过吧。
最终编译出的文件列表

mnn 编译转换工具_第2张图片
原文链接:https://blog.csdn.net/qq_36810544/article/details/105856975

 

 

编译protobuf

参考链接:https://github.com/protocolbuffers/protobuf/blob/master/cmake/README.md

x64

  1. 执行以下代码
cd /path/to/protobuf/vsprojects
cmake -G "Visual Studio 15 2017 Win64" -Dprotobuf_MSVC_STATIC_RUNTIME=ON -Dprotobuf_BUILD_TESTS=OFF ../cmake
  1. 使用VS打开工程并生成解决方案,把Debug产物复制到/path/to/protobuf/vsprojects/x64/Debug下, 把Release产物复制到/path/to/protobuf/vsprojects/x64/Release下。
  2. 将/path/to/protobuf添加至Protobuf_SRC_ROOT_FOLDER环境变量

x86

  1. 执行以下代码
cd /path/to/protobuf/vsprojects
cmake -G "Visual Studio 15 2017" -Dprotobuf_MSVC_STATIC_RUNTIME=ON -Dprotobuf_BUILD_TESTS=OFF ../cmake
  1. 使用VS打开工程并生成解决方案,把Debug产物复制到/path/to/protobuf/vsprojects/Debug下, 把Release产物复制到/path/to/protobuf/vsprojects/Release下。
  2. 将/path/to/protobuf添加至Protobuf_SRC_ROOT_FOLDER环境变量

编译模型转换工具

cd MNN目录

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

 

编译产物

MNNConvert : 模型转换工具

MNNDump2Json : 将MNN模型文件打印为类json文件的工具

常见问题

  1. 如果报错说link error: can not writing MNNConvert.exe, cd到/path/to/MNN/tools/converter/build, 再运行下ninja就好了

cmake -G "Ninja" -DCMAKE_BUILD_TYPE=Release ..

Could NOT find Protobuf (missing: Protobuf_LIBRARIES) (found version
  "3.12.2")

不知道为啥?有build_win的,那一版,就可以

直接建vsprojects,再projects下编译的不行,原因是文件夹名字写错了

你可能感兴趣的:(onnx,c++基础)