python – 将Keras模型集成到TensorFlow中

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 将Keras模型集成到TensorFlow中，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含3537字，纯文字阅读大概需要6分钟。

内容图文

我试图在TensorFlow代码中使用预先训练的Keras模型,如this Keras blog post第II节：使用带有TensorFlow的Keras模型中所述.

我想使用Keras中提供的预先训练的VGG16网络从图像中提取卷积特征图,并在其上添加我自己的TensorFlow代码.所以我做到了这一点：

import tensorflow as tf
from tensorflow.python.keras.applications.vgg16 import VGG16, preprocess_input
from tensorflow.python.keras import backend as K

# images = a NumPy array containing 8 images

model = VGG16(include_top=False, weights='imagenet')
inputs = tf.placeholder(shape=images.shape, dtype=tf.float32)
inputs = preprocess_input(inputs)
features = model(inputs)

with tf.Session() as sess:
    K.set_session(sess)
    output = sess.run(features, feed_dict={inputs: images})
    print(output.shape)

但是,这给了我一个错误：

FailedPreconditionError: Attempting to use uninitialized value block1_conv1_2/kernel
     [[Node: block1_conv1_2/kernel/read = Identity[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"](block1_conv1_2/kernel)]]
     [[Node: vgg16_1/block5_pool/MaxPool/_3 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_132_vgg16_1/block5_pool/MaxPool", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]()]]

相反,如果我在运行网络之前运行初始化程序操作：

with tf.Session() as sess:
    K.set_session(sess)
    tf.global_variables_initializer().run()
    output = sess.run(features, feed_dict={inputs: images})
    print(output.shape)

然后我得到预期的输出：

(8, 11, 38, 512)

我的问题是,运行tf.global_variables_initializer()后,变量是随机初始化还是使用ImageNet权重？我问这个是因为上面引用的博客文章没有提到在使用预训练的Keras模型时需要运行初始化程序,事实上它让我感到有点不安.

我怀疑它确实使用了ImageNet权重,并且只需要运行初始化程序,因为TensorFlow需要显式初始化所有变量.但这只是猜测.

解决方法:

TLDR

使用Keras时,

>如果可以的话,避免使用Session(本着不可知的Keras的精神)
>否则通过tf.keras.backend.get_session使用Keras处理的会话.
>使用Keras的set_session进行高级用途(例如,当您需要进行性能分析或设备放置时)并且在程序的早期阶段 – 与“纯”Tensorflow中的常规做法和良好用法相反.

更多关于这一点

必须先初始化变量才能使用它们.实际上,它比它更微妙：变量必须在它们被使用的会话中初始化.我们来看看这个例子：

import tensorflow as tf

x = tf.Variable(0.)

with tf.Session() as sess:
    tf.global_variables_initializer().run()
    # x is initialized -- no issue here
    x.eval()

with tf.Session() as sess:
    x.eval()
    # Error -- x was never initialized in this session, even though
    # it has been initialized before in another session

因此,您的模型中的变量不会被初始化,因为您在sess之前创建模型,这不足为奇.

但是,VGG16不仅为模型变量(使用tf.global_variables_initializer调用的模型变量)创建初始化操作,而且实际上会调用它们.问题是,哪个会话？

好吧,既然你没有在你构建模型时存在,Keras为你创建了一个默认模式,你可以使用tf.keras.backend.get_session()恢复.现在使用此会话按预期工作,因为变量在此会话中初始化：

with tf.keras.backend.get_session() as sess:
    K.set_session(sess)
    output = sess.run(features, feed_dict={inputs: images})
    print(output.shape)

请注意,您也可以创建自己的Session并通过keras.backend.set_session将其提供给Keras – 这正是您所做的.但是,正如这个例子所示,Keras和TensorFlow有不同的心态.

TensorFlow用户通常首先构建图形,然后在冻结图形之后实例化会话.

Keras与框架无关,并且在构造阶段之间没有这种内置的区别 – 特别是,我们在这里了解到Keras可以在图形构建期间很好地实例化Session.

出于这个原因,当使用Keras时,我建议不要自己管理tf.Session,而是依赖于tf.keras.backend.get_session,如果你需要处理需要tf.Session的TensorFlow特定代码.

内容总结

以上是互联网集市为您收集整理的python – 将Keras模型集成到TensorFlow中全部内容，希望文章能够帮你解决python – 将Keras模型集成到TensorFlow中所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

---