Skip to content

'Tensor' object has no attribute 'ndim' #102

Description

@EnalisUs

why this wrong
%%time

from tf_keras_vis.gradcam_plus_plus import GradcamPlusPlus

Create GradCAM++ object

gradcam = GradcamPlusPlus(model,
model_modifier=model_modifier,
clone=False)
score = CategoricalScore([1, 1,1])

Generate heatmap with GradCAM++

cam = gradcam(score, images, penultimate_layer=-1)

Since v0.6.0, calling normalize() is NOT necessary.

cam = normalize(cam)

Render

f, ax = plt.subplots(nrows=1, ncols=3, figsize=(12, 4))
for i, title in enumerate(image_titles):
heatmap = np.uint8(cm.jet(cam[i])[..., :3] * 255)
ax[i].set_title(title, fontsize=16)
ax[i].imshow(images[i])
ax[i].imshow(heatmap, cmap='jet', alpha=0.5)
ax[i].axis('off')
plt.tight_layout()
plt.savefig('/content/gradcam_plus_plus.png')
plt.show()

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions