write numpy ndarray to Image

Question

I'm trying to read a binary file (8 bit RGB tuples) in Python, do some conversion on it and then write it as a png image. I'm doing the following:

typeinfo = np.dtype('>i1' ) #read single bytes
data=np.fromfile(("f%05d.txt" %(files[ctr])),dtype=typeinfo)
data=np.reshape(data,[linesperfile,resX,3]) #reshape to size/channels

If I display the type information of data it says:

<type 'numpy.ndarray'>
(512L, 7456L, 3L)

Then I do some manipulation on the image (in-place), afterwards I want to write the Image to a file. Currently I use:

import PIL.Image as im
svimg=im.fromarray(data)
svimg.save(("im%05d"%(fileno)),"png")

but it keeps giving me the following error:

line 2026, in fromarray
raise TypeError("Cannot handle this data type")
TypeError: Cannot handle this data type

Any ideas how to do this?

Have you seen stackoverflow.com/questions/4711880/…? It seems that it can't handle RGB in that way. — matsjoyce
– matsjoyce, Commented Dec 23, 2014 at 15:22

Alex Martelli · Accepted Answer · 2014-12-23 15:28:10Z

57

Image needs unsigned bytes, i1 means signed bytes. If the sign is irrelevant (all values between 0 and 127), then this will work:

svimg=im.fromarray(data.astype('uint8'))

If you need the full range 0-255, you should use 'uint8' throughout.

answered Dec 23, 2014 at 15:28

Alex Martelli

887k175 gold badges1.3k silver badges1.4k bronze badges

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

1 Comment

Meng Lee Over a year ago

This is really helpful!

Collectives™ on Stack Overflow

write numpy ndarray to Image

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related