TensorFlow/TensorFlow 保存 pb 模型

TensorFlow/TensorFlow 保存 pb 模型

我们在保存 TensorFlow 的模型时,一般都是把模型保存为 ckpt 文件。

但是如果把模型迁移到手机上,那么就要把模型保存为 pb 文件。模型保存为 pb 文件时候,模型的变量都会变成固定的,从而会大大减小模型的大小,适合在手机端运行。

这里来讲下如何把模型保存为 pb 文件,以及如何从 pb 文件中加载模型。

阅读更多
TensorFlow/[TensorFlow 学习笔记 ] 8 时间序列 2 - 神经网络方法
TensorFlow/[TensorFlow 学习笔记 ] 7 时间序列 1 - 统计学方法
TensorFlow/[TensorFlow 学习笔记 ] 6 RNN 实现文本生成
TensorFlow/[TensorFlow 学习笔记 ] 5 使用 RNN 进行文本分类
TensorFlow/[TensorFlow 学习笔记 ] 4 文本数据预处理
TensorFlow/[TensorFlow 学习笔记 ] 3 计算机视觉 2-数据增强,迁移学习,以及 dropout

TensorFlow/[TensorFlow 学习笔记 ] 3 计算机视觉 2-数据增强,迁移学习,以及 dropout

图片二分类

在上一篇文章中,你学习了如何创建卷积神经网络,以及 ImageGenerator 来创建并训练一个完整的深度学习模型。但之前使用的数据都是计算机生成的,这些数据中的物体都在图片中心,图片宽高一样。而真实世界的图片数据是多种多样的。

在这篇文章中,我们来看下如何处理真实世界的数据。

阅读更多
TensorFlow/[TensorFlow 学习笔记 ] 2 计算机视觉 1-CNN

TensorFlow/[TensorFlow 学习笔记 ] 2 计算机视觉 1-CNN

在上一篇文章中,你学习了如何创建一个神经网络,来学习你要处理的问题。但上一篇文章中的神经网络处理的是非常简单的问题 \(y=2 \times x-1\)。现在我们来看下更加实际,并且也更加困难的问题。在这个问题中,我们使用的数据包含 10 种不同类型的数据,我们要训练一个神经网络,来识别衣服的类型。

阅读更多
TensorFlow/[TensorFlow 学习笔记 ] 1 机器学习入门