玩转Jetson nano系列（3）：实现tensorflow转ncnn（ckpt-> pb param bin)

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/hanzy88/article/details/100084814

主要代码及使用步骤已上传github，觉得有帮助的话麻烦给个星，谢谢~

ckpt转为Pb文件

1. 首先训练模型保存为ckpt文件。
2. 将ckpt文件转为protxt文件（获取各部分输出结点）
3. 将protxt文件转为pb文件，详情请见：

https://github.com/hanzy88/ckpt2ncnn

4. 通过文件中的tensorflow2ncnn文件可以将生成的pb文件转换成需要的param, bin文件；

tensorflow2ncnn

如果需要添加目前不支持的层，可以参考目前我改的转换源码，详情请见：

https://github.com/hanzy88/tensorflow2ncnn