概述
导读
上一篇文章中我们介绍了如何在windows上配置一个ncnn的环境,这篇文章我们来介绍如何将mxnet
、caffe
、TensorFlow
、pytorch
、keras
等深度学习框架训练的模型转换为ncnn的模型,以便于我们后面部署在各种生产环境上。这里主要还是以windows为例
ncnn转换工具
在我们编译好ncnn之后,在ncnn目录里面有一个tools目录,里面包含了一些模型的转换工具
目前ncnn官方只提供了mxnet
、onnx
、caffe
模型的转换,对于其它框架训练的模型需要使用第三方工具或先将模型转为onnx
之后,再将其转换为ncnn模型
注意: 在使用ncnn转换模型的时候可能会遇到一些不支持的操作,遇到这种情况可能需要自己修改模型的架构重新训练模型
或者修改ncnn源码然后重新编译ncnn
,建议采用前者。对于大多数的操作ncnn都是支持,也不用太担心这个问题。
onnx介绍
onnx
(Op
最后
以上就是火星上背包为你收集整理的使用ncnn对模型文件的格式进行转换的全部内容,希望文章能够帮你解决使用ncnn对模型文件的格式进行转换所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复