主要代码及使用步骤已上传github,觉得有帮助的话麻烦给个星,谢谢~
ckpt转为Pb文件
- 首先训练模型保存为ckpt文件。
- 将ckpt文件转为protxt文件(获取各部分输出结点)
- 将protxt文件转为pb文件,详情请见:
https://github.com/hanzy88/ckpt2ncnn
- 通过文件中的tensorflow2ncnn文件可以将生成的pb文件转换成需要的param, bin文件;
tensorflow2ncnn
如果需要添加目前不支持的层,可以参考目前我改的转换源码,详情请见: