在学习《Deep Learning with Python》第五章深度学习用于计算机视觉
5.4.3可视化类激活的热力图
在Tensorflow2.0环境中运行一下代码时
grads = K.gradients(african_elephant_output, last_conv_layer.output)[0]
替换成
grads = tf.keras.backend.gradients(african_elephant_output, last_conv_layer.output

在Tensorflow 2.0环境下,尝试使用tf.gradients进行梯度计算时会遇到不支持的错误,应当改用tf.GradientTape。这一改变是由于Tensorflow从静态图模式转向了Eager Execution。为了解决这个问题,可以参考Stack Overflow上的解决方案,通过tf.GradientTape实现类激活热力图的可视化。在《Deep Learning with Python》第五章的深度学习用于计算机视觉部分,特别是在5.4.3节中,有相关的代码示例。
最低0.47元/天 解锁文章
712





