tensorflow实现迁移学习实例
来源:互联网 发布:数值分析教材 知乎 编辑:程序博客网 时间:2024/06/05 04:15
tensorflow实现迁移学习实例
本文主要是总结利用tensorflow实现迁移学习的基本步骤。
所谓迁移学习,就是将上一个问题上训练好的模型通过简单的调整使其适用于一个新的问题。比如说,我们可以保留训练好的Inception-v3模型中所有的参数,只替换最后一层全连接层。在最后一层全连接层之前的网络称之为瓶颈层(bottleneck)。
持久化
首先需要简单介绍下tensorflow中的持久化:在tensorflow中提供了一个非常简单的API来保存和还原一个神经网络模型,这个API就是tf.train.Saver类。当采用该方法保存时会生成三个文件,一个文件是model.ckpt.meta,它保存了Tensorflow计算图的结构;第二个文件是model.ckpt,它保存了程序中每一个变量的取值;最后一个文件是checkpoint文件,这个文件中保存了一个目录下所有模型文件列表。
保存图
init_op = tf.initialize_all_variables()with tf.Session() as sess: sess.run(init_op) saver.save(sess, "model.ckpt")
加载图
saver = tf.train.import_meta_graph("model.ckpt.meta")with tf.Session() as sess: saver.restore(sess, "model.ckpt")
迁移学习
第一步: 读取加载已经训练好的模型
在inception-v3模型代表瓶颈层结果的张量名称是’pool3/_reshape:0’,图像输入张量对应的名称’DecodeJpeg/contents:0’
BOTTLENECK_TENSOR_NAME = 'pool_3/_reshape:0'JPEG_DATA_TENSOR_NAME = 'DecodeJpeg/contents:0'#读取已经训练好的模型 with gfile.FastGFile(os.path.join(MODEL_DIR, MODEL_FILE), 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) bottleneck_tensor, jpeg_data_tensor = tf.import_graph_def(graph_def, return_elements=[BOTTLENECK_TENSOR_NAME, JPEG_DATA_TENSOR_NAME])
第二步:利用读取的模型,定义新的神经网络输入,这个输入就是新的图片经过Inception-v3模型前向传播到达瓶颈层的取值,是一种特征提取过程。
def run_bottlenect_on_images(sess, image_data, image_data_tensor, bottlenect_tensor): bottlenect_values = sess.run(bottlenect_tensor, {image_data_tensor: image_data}) # 经过卷积网络处理后的是一个思维数组,压缩成一个特征,一维向量输出 bottlenect_values = np.squeeze(bottlenect_values) return bottlenect_values
该过程实际上利用获取的tensor计算图片的特征向量,完成特征提取的过程。
第三步:利用获取的图像的特征向量完成接下来的任务(比如分类)
以上是仅关键代码。
阅读全文
0 2
- tensorflow实现迁移学习实例
- tensorflow实现迁移学习
- tensorflow实现迁移学习
- tensorflow实现迁移学习
- 迁移学习的TensorFlow实现
- 用tensorflow实现迁移学习的一个例子
- 理解深层神经网络中的迁移学习及TensorFlow实现
- 度学习实践:如何使用Tensorflow实现快速风格迁移?
- Tensorflow实例:实现AlexNet
- Tensorflow实例:实现VGG
- Tensorflow实例:实现ResNet
- Tensorflow实例:实现深度强化学习--策略网络
- Tensorflow实现照片风格的迁移
- 用tensorflow实现usps和mnist数据集的迁移学习
- 用tensorflow实现usps和mnist数据集的迁移学习
- Tensorflow-2: 迁移学习程序编写
- Tensorflow基础:卷积神经网络迁移学习
- 【TensorFlow】迁移学习(使用Inception-v3)
- angular-resource的url转义问题
- UE4的AI学习(1)——基本概念
- touch
- protoc 生成C++代码
- Could not open the editor: Invalid thread access报错
- tensorflow实现迁移学习实例
- 最近都在说移民火星,这些黑科技你了解吗?
- Spring定时任务的几种实现
- yii2 的Urlmanager 404问题
- OKHTTP使用 保存session
- 一张图看学霸, 杭州云栖大会前13位顶尖科学家聚会
- Swing JComponent类
- Java中使用根据模板拼接字符串的两种方式
- Python 递归,遍历文件夹