ImageClassificationExplainer: AttributeError: ndim when trying to visualize.

[mollie9523]

Hi,

I am trying to run the notebook that you provided but when it comes to visualizing, there is an attribute error.

How can I solve this?

[oskwys]

I have the same problem when playing with the exemplary code

`from transformers import AutoFeatureExtractor, AutoModelForImageClassification
from transformers_interpret import ImageClassificationExplainer
from PIL import Image
import requests

model_name = "google/vit-base-patch16-224"
model = AutoModelForImageClassification.from_pretrained(model_name)
feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)

image_link = "https://imagesvc.meredithcorp.io/v3/mm/image?url=https%3A%2F%2Fstatic.onecms.io%2Fwp-content%2Fuploads%2Fsites%2F47%2F2020%2F08%2F16%2Fgolden-retriever-177213599-2000.jpg"

image = Image.open(requests.get(image_link, stream=True).raw)

image_classification_explainer = ImageClassificationExplainer(model=model, feature_extractor=feature_extractor)

image_attributions = image_classification_explainer(
image
)

print(image_attributions.shape)

image_classification_explainer.visualize(
method="heatmap",
side_by_side=True,
outlier_threshold=0.03

)`

[MohieEldinMuhammad]

Do you know the solution to this error?

[lukassoukup]

Hi, I encountered the same error, it seems that transformers library doesn't work with PIL. Changing line 69 in transformers_interpret/explainers/vision/image_classification.py to this:
self.feature_extractor.resize(np.array(self._image), size=(attributions_t.shape[0], attributions_t.shape[1]))
worked in my case.