pytorch训练模型移植至ncnn
pytorch训练模型移植至ncnn
第一步是pytorch模型往caffe模型的转换
采用的工具是网路公开的代码，，此时会生成prototxt和caffemodel两个新文件。
第二部是往ncnn迁移
ncnn安装 安装好之后，进行相应的转换 1）首先，上一步生成的mobilenetv3.prototxt和mobilenetv3.caffemodel文件是比价旧的版本，这时候需要转换为新的版本，这时候要用到caffe自带的转换工具
upgrade_net_proto_text [old prototxt] [new prototxt] upgrade_net_proto_binary [old caffemodel] [new caffemodel]
其中这两个工具的路径在 caffe/.build_release/tools路径下面，转换完之后会生成两个全新的caffemodel和prototxt
2）接下来修改prototxt文件头，这一步是必须的。 3）使用caffe2ncnn 将caffe模型转化为ncnn模型
./caffe2ncnn mobilenetv3_9_29_new.prototxt mobilenetv3_9_29_new.caffemodel mobilenetv3_9_29_new.param mobilenetv3_9_29_new.bin
此时生成bin和param两个文件
4）加下来就在cpp里面去调用就可以啦 在ncnn/src目录下面添加想添加的源文件，在ncnn/examples下面添加处理文件，然后在ncnn/build目录下make -j4即可生成，注意修改对应的CMakeList.txt文件
下一篇：
解决Tomcat中文乱码问题