本文记录一下模型部署相关的MindSporeLite模型转换过程:mindir模型转ms模型
这里的ms模型是适用于端侧部署的,所以这个转换可以看成是部署的第一步。
环境
64位X86 ubuntu 16.04
MindSpore 1.1.0
clone代码
因为需要在本地编译并生成模型转换与推理相关的一系列文件,先下载MindSpore代码。
这里是从r1.1这个分支下载的。
unseenme@v100:~/study/github$ git clone -b r1.1 --depth=1 https://github.com/mindspore-ai/mindspore.git
访问Github困难的小伙伴们可以从Gitee下载。
编译
下载完成后,进入目录,通过下面的脚本与参数,进行一键编译。
这个脚本真的是非常方便,这里手动点赞!
unseenme@v100:~/study/github/mindspore$ bash build.sh -I x86_64
编译需要一定的时间,最后完成的时候有如下提示。
---------------- mindspore lite: build success ----------------
编译成功,在output目录下生成四个文件。
- mindspore-lite-1.1.0-converter-linux-x64.tar.gz
- mindspore-lite-1.1.0-converter-linux-x64.tar.gz.sha256
- mindspore-lite-1.1.0-inference-linux-x64.tar.gz
- mindspore-lite-1.1.0-inference-linux-x64.tar.gz.sha256
如下图所示。
其中,带有converter的包里是模型转换相关文件,
带有inference的包里是模型推理相关文件。
解压converter包,看看里面都有什么。
unseenme@v100:~/study/github/msmaster/output$ tar -xf mindspore-lite-1.1.0-converter-linux-x64.tar.gz
unseenme@v100:~/study/github/msmaster/output$ tree ./mindspo