Invalid dimension for image data in plt.imshow()

Question

I am using mnist dataset for training a capsule network in keras background. After training, I want to display an image from mnist dataset. For loading images, mnist.load_data() is used. The data is stored as (x_train, y_train),(x_test, y_test). Now, for visualizing image, my code is as follows:

img_path = x_test[1]  
print(img_path.shape)
plt.imshow(img_path)
plt.show()

The code gives output as follows:

(28, 28, 1)

and the error on plt.imshow(img_path) as follows:

TypeError: Invalid dimensions for image data

How to show image in png format. Help!

You should remove the redundant channel dimension. Try `plt.imshow(np.squeeze(img_path))` — sdcbr, Feb 13 '19 at 07:13
Does this answer your question? [Showing an image with pylab.imshow()](https://stackoverflow.com/questions/15008045/showing-an-image-with-pylab-imshow) — iacob, Mar 24 '21 at 21:35

shantanu pathak · Answer 1 · 2020-01-18T03:43:47.127

As per the comment of @sdcbr using np.sqeeze reduces unnecessary dimension. If image is 2 dimensions then imshow function works fine. If image has 3 dimensions then you have to reduce extra 1 dimension. But, for higher dim data you will have to reduce it to 2 dims, so np.sqeeze may be applied multiple times. (Or you may use some other dim reduction functions for higher dim data)

import numpy as np  
import matplotlib.pyplot as plt
img_path = x_test[1]  
print(img_path.shape)
if(len(img_path.shape) == 3):
    plt.imshow(np.squeeze(img_path))
elif(len(img_path.shape) == 2):
    plt.imshow(img_path)
else:
    print("Higher dimensional data")

score 3 · Answer 2 · edited May 13 '20 at 09:33

3

Example:

plt.imshow(test_images[0])

TypeError: Invalid shape (28, 28, 1) for image data

Correction:

plt.imshow((tf.squeeze(test_images[0])))

Number 7

edited May 13 '20 at 09:33

Simas Joneliunas

2,890
20
28
35

answered May 13 '20 at 08:00

Deepak

31
1

score 2 · Accepted Answer · answered Dec 25 '19 at 13:32

2

You can use tf.squeeze for removing dimensions of size 1 from the shape of a tensor.

plt.imshow( tf.shape( tf.squeeze(x_train) ) )

Check out TF2.0 example

answered Dec 25 '19 at 13:32

Parth Patel

68
1
9

score 0 · Answer 4 · edited Dec 28 '21 at 17:05

0

matplotlib.pyplot.imshow() does not support images of shape (h, w, 1). Just remove the last dimension of the image by reshaping the image to (h, w): newimage = reshape(img,(h,w)).

edited Dec 28 '21 at 17:05

jeffhale

3,759
7
40
56

answered Dec 05 '20 at 07:42

Sikander SD

1

Invalid dimension for image data in plt.imshow()

4 Answers4