摘要:
本博客实现将自己训练保存的ckpt模型转换为pb文件,该方法适用于任何ckpt模型,当然你需要确定ckpt模型输入/输出的节点名称。 目录 tensorflow实现将ckpt转pb文件 一、CKPT 转换成 PB格式 二、 pb模型预测 三、源码下载和资料推荐 1、训练方法 2、本博客Github地 阅读全文
摘要:
【摘要】 笔者最近因为工作需要将TensorFlow训练模型迁移到晟腾芯片平台上离线推理,所以需要将ckpt或者h5模型冻结成pb,再利用ATC模型转换工具将pb转为离线模型om文件,期间遇到一些问题和坑,总结一下,供大家参考。1.Tensorflow1.x训练好的模型Ckpt文件:DB_resne 阅读全文