微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

TensorFlow如何计算vgg19.preprocess_input的梯度?

如何解决TensorFlow如何计算vgg19.preprocess_input的梯度?

我正在关注neural style transfer上的教程。通过最小化相对于图像(用内容图像初始化)的损失函数来完成样式转换。以下代码让我感到困惑:

preprocessed_input = tf.keras.applications.vgg19.preprocess_input(inputs)

,它是call类中StyleContentModel方法的一部分。 TensorFlow如何知道此操作的梯度?我已经使用模块get_gradient_function中的tensorflow.python.framework.ops检查了此操作是否具有梯度函数,据我所知没有。

解决方法

这与模型或渐变无关。此功能的作用是缩放输入图像,使像素在-1到+1的范围内。这是用于迁移学习中的许多模型(例如VGG和MobileNet)的普遍要求。如果使用ImageDataGenerator,则它具有一个参数preprocessing_function,生成器会调用该参数来预处理图像。确保是否对训练图像进行了预处理,对测试图像和验证图像进行了相同的操作。

,

这非常简单,该函数在内部使用可微分的符号张量运算。 TensorFlow可以通过内部使用TensorFlow操作的函数来计算梯度,无需为每个函数手动定义梯度。

您可以通过查看该函数here的代码来进行确认,特别是如果您查看的_preprocess_symbolic_function here使用的是普通标量运算和Keras后端函数( tf.keras中的TensorFlow函数。

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

相关推荐


Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其他元素将获得点击?
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。)
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbcDriver发生异常。为什么?
这是用Java进行XML解析的最佳库。
Java的PriorityQueue的内置迭代器不会以任何特定顺序遍历数据结构。为什么?
如何在Java中聆听按键时移动图像。
Java“Program to an interface”。这是什么意思?