Mercurial > ift6266

comparison pycaptcha/transformations.py~ @ 87:4775b4195b4b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
code pour la generation de captchas
author goldfinger
date Thu, 11 Feb 2010 05:09:46 -0500
parents
children
comparison
equal deleted inserted replaced
86:b3d76ebf2fac 87:4775b4195b4b
1
2 import Numeric, Image
3 """ Transforme une image PIL en objet numpy.array et vice versa"""
4
5
6 def image2array(im):
7 """ image vers array numpy"""
8 if im.mode not in ("L", "F"):
9 raise ValueError, "can only convert single-layer images"
10 if im.mode == "L":
11 a = Numeric.fromstring(im.tostring(), Numeric.UnsignedInt8)
12 else:
13 a = Numeric.fromstring(im.tostring(), Numeric.Float32)
14 a.shape = im.size[1], im.size[0]
15 return a
16
17 def array2image(a):
18 """ array numpy vers image"""
19 if a.typecode() == Numeric.UnsignedInt8:
20 mode = "L"
21 elif a.typecode() == Numeric.Float32:
22 mode = "F"
23 else:
24 raise ValueError, "unsupported image mode"
25 return Image.fromstring(mode, (a.shape[1], a.shape[0]), a.tostring())