convert multidimensional numpy array to image

Question

I have a numpy array(trainData.npy) for image processing. It contains 2000 images, garyscale and height 450 , width 600.

Train images shape:(2000, 1, 450, 600)

I'm looking for a way to plot or show one of these images. I've used this code but i'v got TypeError: Invalid dimensions for image data error/

import numpy as np
import matplotlib.pyplot as plt

img = img_train[0]
plt.imshow(img)
plt.show()

I've already done. img_train[0].shape the answer is (1, 450, 600) — Sarah K
– Sarah K, Commented Jun 1, 2018 at 15:14
image show can show two dimensional arrays. img = img_train[0,0] — anishtain4
– anishtain4, Commented Jun 1, 2018 at 17:08

mzoll · Accepted Answer · 2018-06-04 08:39:22Z

1

Just reshape your image:

import numpy as np
import matplotlib.pyplot as plt

img = img_train[0] #img has dim (ncolor=1, nlines=450, nrows=600)
img = reshape(450, 600) #img has dim (450, 600)
#img = img / img.max #if you need rescaling of greyscale to be in [0..1]
plt.imshow(img)
plt.show()

P.S.: I personally find it a bit more intuitive if pictures are dimensional-ordered (nlines, nrows, ncolorchanel) than your choice (ncolorchanel, nlines, nrows)

edited Jun 4, 2018 at 8:39

answered Jun 4, 2018 at 8:33

mzoll

4873 silver badges11 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Sarah K Over a year ago

img = img.reshape(450, 600) it worked great. Thanks!

Collectives™ on Stack Overflow

convert multidimensional numpy array to image

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related