Store a numpy.ndarray as an image, and then save the pixel values

Question

I am currently looking for at in which i can store a numpy.ndarray as an image, then save that image, and extract the pixel values of the image into a numpy.ndarray. The dimension of the numpy.ndarray with the pixel values should be the same as the numpy.ndarray what was used to create the plot.

I tried something like this:

def make_plot_store_data(name, data):
    plt.figure()
    librosa.display.specshow(data.T,sr=16000,x_axis='frames',y_axis='mel',hop_length=160,cmap=cm.jet)
    plt.savefig(name+".png")
    plt.close()

    convert = plt.get_cmap(cm.jet)
    numpy_output_interweawed = convert(data.T)

The first image looks like this: The second image looks like this:

Why is this so messed up?

It's easier to help when you provide a Minimum, Complete, and Verifiable example. Also, it's not clear why your ndarray would change shape just because it's rendered as an image. See my answer below for a working example. — andrew_reece
– andrew_reece, Commented Apr 27, 2017 at 7:16
Also, your convert() maps scalars to RGBA, which will return an [N, N, 4] matrix, given an [N x N] input array. See Colormap docs for more. — andrew_reece
– andrew_reece, Commented Apr 27, 2017 at 7:52
Same question as this one: stackoverflow.com/questions/43646838/… — ImportanceOfBeingErnest
– ImportanceOfBeingErnest, Commented Apr 27, 2017 at 8:38

andrew_reece · Accepted Answer · 2017-04-27 07:42:04Z

Here's one approach that takes a 512x512 ndarray, displays it as an image, stores it as an image object, saves an image file, and generates a normalized pixel array of the same shape as the original.

import numpy as np

# sample numpy array of an image
from skimage import data
camera = data.camera()
print(camera.shape) # (512, 512)

# array sample
print(camera[0:5, 0:5])

[[156 157 160 159 158]
 [156 157 159 158 158]
 [158 157 156 156 157]
 [160 157 154 154 156]
 [158 157 156 156 157]]

# display numpy array as image, save as object
img = plt.imshow(camera)

# save image to file
plt.savefig('camera.png')

# normalize img object pixel values between 0 and 1
normed_pixels = img.norm(camera)

# normed_pixels array has same shape as original 
print(normed_pixels.shape) # (512, 512)

# sample data from normed_pixels numpy array 
print(normed_pixels[0:5,0:5])

[[ 0.61176473  0.6156863   0.627451    0.62352943  0.61960787]
 [ 0.61176473  0.6156863   0.62352943  0.61960787  0.61960787]
 [ 0.61960787  0.6156863   0.61176473  0.61176473  0.6156863 ]
 [ 0.627451    0.6156863   0.60392159  0.60392159  0.61176473]
 [ 0.61960787  0.6156863   0.61176473  0.61176473  0.6156863 ]]

You might consider looking into the skimage module, in addition to standard pyplot methods. There are a bunch of image manipulation methods there, and they're all built to play nice with numpy. Hope that helps.

Collectives™ on Stack Overflow

Store a numpy.ndarray as an image, and then save the pixel values

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related