如何使用TensorFlow搭建网络并导出、运行模型实例?
- 内容介绍
- 文章标签
- 相关推荐
本文共计2582个文字,预计阅读时间需要11分钟。
概述:之前自己都利用别人的搭建好的工程,修改过来用,很少把模型搭建、导出、加载模型运行走一遍,搞懂了这个事情也不是那么简单的。搭建模型和导出模型+参考Ten。
概述
以前自己都利用别人搭好的工程,修改过来用,很少把模型搭建、导出模型、加载模型运行走一遍,搞了一遍才知道这个事情也不是那么简单的。
搭建模型和导出模型
参考《TensorFlow固化模型》,导出固化的模型有两种方式.
方式1:导出pb图结构和ckpt文件,然后用 freeze_graph 工具冻结生成一个pb(包含结构和参数)
在我的代码里测试了生成pb图结构和ckpt文件,但是没接着往下走,感觉有点麻烦。我用的是第二种方法。
注意我这里只在最后保存了一次ckpt,实际应该在训练中每隔一段时间就保存一次的。
本文共计2582个文字,预计阅读时间需要11分钟。
概述:之前自己都利用别人的搭建好的工程,修改过来用,很少把模型搭建、导出、加载模型运行走一遍,搞懂了这个事情也不是那么简单的。搭建模型和导出模型+参考Ten。
概述
以前自己都利用别人搭好的工程,修改过来用,很少把模型搭建、导出模型、加载模型运行走一遍,搞了一遍才知道这个事情也不是那么简单的。
搭建模型和导出模型
参考《TensorFlow固化模型》,导出固化的模型有两种方式.
方式1:导出pb图结构和ckpt文件,然后用 freeze_graph 工具冻结生成一个pb(包含结构和参数)
在我的代码里测试了生成pb图结构和ckpt文件,但是没接着往下走,感觉有点麻烦。我用的是第二种方法。
注意我这里只在最后保存了一次ckpt,实际应该在训练中每隔一段时间就保存一次的。

