在Google Colab生成GradCAM热力图时遇EagerTensor无astype属性错误

问题场景 #

在Google Colab中使用tf-keras-vis库生成激活最大化图像时，执行image = activation[0].astype(np.uint8)时触发错误：

AttributeError: EagerTensor object has no attribute 'astype'.

解决方法 #

先将EagerTensor转换为numpy数组，再调用astype()方法，同时确保已导入numpy库。修改后的关键代码如下：

import numpy as np  # 需补充导入numpy

# ... 其余代码保持不变 ...

activation = activation_maximization(loss, callbacks=[Print(interval=100)])
image = activation[0].numpy().astype(np.uint8)  # 先转numpy数组再做类型转换

f, ax = plt.subplots(figsize=(10, 5), subplot_kw={'xticks': [], 'yticks': []})
ax.imshow(image)
plt.show()

原因说明 #

tf-keras-vis的ActivationMaximization返回的结果是TensorFlow的EagerTensor对象，而非numpy数组。astype()是numpy数组的专属方法，EagerTensor不支持该方法，因此需要先用.numpy()将其转换为numpy数组，再执行类型转换。

内容的提问来源于stack exchange，提问作者upaf