[Image-SIG] help for conversion of NUMARRAY to PIL object
Bob Klimek
klimek at grc.nasa.gov
Fri Sep 16 14:19:25 CEST 2005
Zhenglong Li wrote:
>Hi, Fredrik,
>
> Thank you very much for your help. But there still exists a problem
>that the code cann't deal with 3-channel RGB image. If I want to read a
>len*width*3 matrix from PIL object into NumArray, and vice versa, what
>should I do?
>
>
Try this.
import numarray as N
import Image
screenLevels = 255.0 # we have 8 bit video cards
def image2array(im, convertType='UInt8'):
"""
Convert PIL image to Numarray array
"""
if im.mode == "L":
a = N.fromstring(im.tostring(), N.UInt8)
a = N.reshape(a, (im.size[1], im.size[0]))
#a.shape = (im.size[1], im.size[0], 1) # alternate way
elif (im.mode=='RGB'):
a = N.fromstring(im.tostring(), N.UInt8)
a.shape = (im.size[1], im.size[0], 3)
else:
raise ValueError, im.mode+" mode not considered"
if convertType == 'Float32':
a = a.astype(N.Float32)
return a
def array2image(a):
"""
Converts array object (numarray) to image object (PIL).
"""
h, w = a.shape[:2]
int32 = N.Int32
uint32 = N.UInt32
float32 = N.Float32
float64 = N.Float64
if a.type()==int32 or a.type()==uint32 or a.type()==float32 or
a.type()==float64:
a = a.astype(N.UInt8) # convert to 8-bit
if len(a.shape)==3:
if a.shape[2]==3: # a.shape == (y, x, 3)
r = Image.fromstring("L", (w, h), a[:,:,0].tostring())
g = Image.fromstring("L", (w, h), a[:,:,1].tostring())
b = Image.fromstring("L", (w, h), a[:,:,2].tostring())
return Image.merge("RGB", (r,g,b))
elif a.shape[2]==1: # a.shape == (y, x, 1)
return Image.fromstring("L", (w, h), a.tostring())
elif len(a.shape)==2: # a.shape == (y, x)
return Image.fromstring("L", (w, h), a.tostring())
else:
raise ValueError, "unsupported image mode"
Bob
More information about the Image-SIG
mailing list