Problème pywt - PIL (double/entiers)

Version imprimable

Bonsoir,
J'utilise la lib PIL pour ouvrir et enregistrer des JPEG.
J'utilise aussi Pywt pour faire le transformée en ondelettes discrète.
Le truc, et c'est mon problème, c'est que je sais convertir chaque composante (R,G,B) de mon image, et je souhaite recréer une image réduite de moitié par les ondelettes.
Malheureusement, cette dernière étape ne fonctionne pas, je pense que c'est dû au fait que la fonction pywt.dwt() me renvoie une liste (à priori) de double, et qu'il me faut des int.
Code:

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 R,G,B=im.split(); R=list(R.getdata()); G=list(G.getdata()); B=list(B.getdata()); (RA,RD)=pywt.dwt(R,'db2','sp1') (GA,GD)=pywt.dwt(G,'db2','sp1') (BA,BD)=pywt.dwt(B,'db2','sp1') #C'est normal, la suite, je réagrandi avec erreur RA=pywt.idwt(RA, None, 'db2', 'sp1') GA=pywt.idwt(GA, None, 'db2', 'sp1') BA=pywt.idwt(BA, None, 'db2', 'sp1') comp=(RA,GA,BA) # ICI, ça ne passe plus imgnew = Image.merge('RGB',comp)

Voilà un code complet (copier-collé du tien) qui m'a l'air de fonctionner. J'ai pris lena.jpg ( http://images.google.fr/images?hl=fr...pg&sa=N&tab=wi ) en couleur de taille 512 par 512.
La fonction merge demande en argument un tuple d'images, toi, tu fournissais un couple de listes.
Enfin, au final, j'ai une image qui resemble à la première mais un peu compressé 35 ko contre 67 ko au départ
Code:

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 import pywt import Image im = Image.open('lena.jpg') R,G,B=im.split(); R=list(R.getdata()); G=list(G.getdata()); B=list(B.getdata()); (RA,RD)=pywt.dwt(R,'db2','sp1') (GA,GD)=pywt.dwt(G,'db2','sp1') (BA,BD)=pywt.dwt(B,'db2','sp1') RA=pywt.idwt(RA, None, 'db2', 'sp1') GA=pywt.idwt(GA, None, 'db2', 'sp1') BA=pywt.idwt(BA, None, 'db2', 'sp1') comp=(RA,GA,BA) # ICI, ca ne passe plus comp = (map(int,comp[0]), map(int,comp[1]), map(int,comp[2])) c1 = Image.new('L', (512, 512)) c2 = Image.new('L', (512, 512)) c3 = Image.new('L', (512, 512)) c1.putdata(comp[0]) c2.putdata(comp[1]) c3.putdata(comp[2]) imgnew = Image.merge('RGB',(c1,c2,c3)) imgnew.save('toto.jpg')

15/05/2006, 21h03
progfou

OK, j'ai corrigé mon code en conséquence ;).
Merci !
Sinon, il est possible de faire la différence de deux images ?
15/05/2006, 21h14
Guigui_

oui dans le module ImageChops ( http://python.developpez.com/cours/p...imagechops.php )
y'a la fonction difference ou substract (en fonction de ce que tu cherches)
16/05/2006, 19h48
progfou

Si je veux créer une image deux fois plus grande que mon image originale (ça, je pense savoir), et mettre en haut à gauche mon image originale, je fais comment ?
J'ai vu la méthode Image.box mais pas sûr qu'elle m'aide...
16/05/2006, 20h21
Guigui_

Non, c'est la fonction paste

Citation:

<H3>paste

Citation:

im.paste(image, box)
Pastes another image into this image. The box argument is either a 2-tuple giving the upper left corner, a 4-tuple defining the left, upper, right, and lower pixel coordinate, or None (same as (0, 0)). If a 4-tuple is given, the size of the pasted image must match the size of the region.
If the modes don't match, the pasted image is converted to the mode of this image (see the convert method for details).
im.paste(colour, box)
Same as above, but fills the region with a single colour. The colour is given as a single numerical value for single-band images, and a tuple for multi-band images.
im.paste(image, box, mask)
Same as above, but updates only the regions indicated by the mask. You can use either "1", "L" or "RGBA" images (in the latter case, the alpha band is used as mask). Where the mask is 255, the given image is copied as is. Where the mask is 0, the current value is preserved. Intermediate values can be used for transparency effects.
Note that if you paste an "RGBA" image, the alpha band is ignored. You can work around this by using the same image as both source image and mask.
im.paste(colour, box, mask)
Same as above, but fills the region indicated by the mask with a single colour.

</H3>
16/05/2006, 21h03
progfou
Mmh, j'ai essayé :
Code:

1 2 3 4 width,height=im.size im_big = Image.new('RGB', (width*2, height*2)) im_big.paste(im,None)
Mais il me met :

Citation:

im_big.paste(im,None)
File "C:\Python24\Lib\site-packages\PIL\Image.py", line 1013, in paste
self.im.paste(im, box)
ValueError: images do not match

Voilà un exemple qui marche:

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
 
import Image
im1 = Image.open('lena.jpg') ## lena.jpg image 512x512 couleur
 
size = []
size.append(2 * im1.size[0])
size.append(2 * im1.size[1])
im = Image.new('RGB',size)
size1 = (0,0)
size2 = (im1.size[0], 0)
size3 = (0, im1.size[1])
size4 = im1.size
im.paste(im1, size1)
im.paste(im1, size2)
im.paste(im1, size3)
im.paste(im1, size4)
im.save('toto.jpg')

ton image im est un en RGB ?

16/05/2006, 21h33
progfou

Effectivement, ça fonctionne mieux :).

Merci beaucoup !

Ca aurait été trop beau :).
J'ai ce code :

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
import Image,ImageChops
import pywt
import psyco
 
psyco.full()
im1 = Image.open("lena.jpg");
 
size = []
size.append(2 * im1.size[0])
size.append(2 * im1.size[1])
im = Image.new('RGB',size)
size1 = (0,0)
im.paste(im1, size1)
R,G,B=im1.split();
R=list(R.getdata());
G=list(G.getdata());
B=list(B.getdata());
 
RA=pywt.idwt(R, None, 'db2', 'sp1')
GA=pywt.idwt(G, None, 'db2', 'sp1')
BA=pywt.idwt(B, None, 'db2', 'sp1')
 
comp=(RA,GA,BA)
comp = (map(int,comp[0]), map(int,comp[1]), map(int,comp[2]))
c1 = Image.new('L', size)
c2 = Image.new('L', size)
c3 = Image.new('L', size)
c1.putdata(comp[0])
c2.putdata(comp[1])
c3.putdata(comp[2])
imgnew = Image.merge('RGB',(c1,c2,c3))
imgnew.save('big_lena.jpg')

En théorie, il m'agrandit l'image de base par deux, en pratique, ce n'est pas le cas.

Modifié le code...

17/05/2006, 22h25
Guigui_

1 pièce(s) jointe(s)

moi ton prog me donne ceci (image en piece jointe)
17/05/2006, 22h47
progfou

Je l'ai modifié, j'ai réedité.
17/05/2006, 23h12
Guigui_

Le problème vient de idwt :

j'obtiens:
len(R) = 262144
len(RA) = 512288 (alors que tu voudrais 1024576)
Mais je ne connais pas iwdt. Donc il faut que tu regardes de ce côté.

Sinon, pourquoi utilisez les ondelettes ?

la fonction resize marche très bien.

Code:

im2 = im1.resize((1024,1024))
18/05/2006, 09h21
progfou

Je vais voir avec pywt.
Pourquoi les ondelettes ?
Pour apprendre, car je trouve ça puissant :).

Est-il possible d'obtenir les images RGB sous une forme matricielle ?
Pour pouvoir utiliser ces fonctions :

Code:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import numpy
 
def dwt2(data, wavelet, mode):
   data = numpy.asarray(data, dtype=numpy.float64)
   if len(data.shape) != 2:
       raise ValueError("Expected 2D array")
   wavelet = pywt.Wavelet(wavelet)
 
   H, L = [], []
   for row in data:
       cA, cD = pywt.dwt(row, wavelet, mode)
       L.append(cA)
       H.append(cD)
 
   H = numpy.transpose(H)
   L = numpy.transpose(L)
 
   LL, LH = [], []
   for row in L:
       cA, cD = pywt.dwt(row, wavelet, mode)
       LL.append(cA)
       LH.append(cD)
 
   HL, HH = [], []
   for row in H:
       cA, cD = pywt.dwt(row, wavelet, mode)
       HL.append(cA)
       HH.append(cD)
 
   return map(numpy.array, [LL, LH, HL, HH])
 
def idwt2(LL, LH, HL, HH, wavelet, mode):
   #do the dimension and checking here
 
   L = []
   for rowL, rowH in zip(LL, LH):
       L.append(pywt.idwt(rowL, rowH, wavelet, mode))
 
   H = []
   for rowL, rowH in zip(HL, HH):
       H.append(pywt.idwt(rowL, rowH, wavelet, mode))
 
   L = numpy.transpose(L)
   H = numpy.transpose(H)
 
   data = []
   for rowL, rowH in zip(L, H):
       data.append(pywt.idwt(rowL, rowH, wavelet, mode))
 
   return numpy.array(data, dtype=numpy.float64)

Maintenant, pour pouvoir m'en servir, il me faut un tableau colonnes/lignes.

26/05/2006, 18h26
parp1
salut! Euh je ne suis pas ur de repondre a ta question.Tu peux mettre une image sous forme matricielle a l'aide de la fonction setshape().
Code:

1 2 3 4 5 6 7 import numarray # a essayer avec numpy data=im.getdata() larg,haut=im.size Matrice1D=N.array(data) Matrice2D=Matrice.setshape(larg,haut)
Apres ca marche si tu a un seul composante.
Je pense que tu peux créer une hyper matrice avec trois argument pour setshape.

Voila!
20/09/2006, 12h35
progfou

Je suis désolé de reprendre ce sujet qui est vieux, mais je le reprends aujourd'hui :).
parp1: ton code fonctionne normallement ?
Car N n'est pas connu, et je suppose que c'est Matrice1D.setshape() et non Matrice.setshape() ?
29/09/2006, 00h29
parp1

Dslé
Oula le post qu'on soir de derriere les fagots les longues soirée d'hiver.... lol

Enfait j'ai fait une petit erreur.
Code:

1 2 3 4 5 6 7 import numarray as N# a essayer avec numpy data=im.getdata() larg,haut=im.size Matrice1D=N.array(data) Matrice2D=Matrice1.setshape(larg,haut)
Ca devrait fonctionner... A plus Camarade!