annotate transformations/PermutPixel.py @ 47:3bc75139654a

Ajout d'un petit bruit gaussien sur chaque pixel de l'image.
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Thu, 04 Feb 2010 10:32:07 -0500
parents 48a21d19b8eb
children 9936c4886299
rev   line source
46
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
1 #!/usr/bin/python
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
2 # coding: utf-8
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
3
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
4 '''
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
5 Un echange de pixels est effectue entre certain pixels choisit aleatoirement
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
6 et un de ses 4 voisins, tout aussi choisi aleatoirement.
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
7
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
8 Le nombre de pixels permutes est definit pas complexity*1024
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
9
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
10 Sylvain Pannetier Lebeuf dans le cadre de IFT6266, hiver 2010
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
11
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
12 '''
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
13
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
14 import numpy
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
15 import random
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
16
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
17 class PermutPixel():
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
18
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
19 def __init__(self):
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
20 self.nombre=10 #Le nombre de pixels a permuter
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
21 self.proportion=0.3
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
22
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
23 def get_settings_names(self):
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
24 return ['nombre']
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
25
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
26 def regenerate_parameters(self, complexity):
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
27 self.proportion=float(complexity)
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
28 self.nombre=int(256*self.proportion)*4 #Par multiple de 4 (256=1024/4)
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
29 return self._get_current_parameters()
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
30
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
31 def _get_current_parameters(self):
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
32 return []
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
33
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
34 def get_parameters_determined_by_complexity(self, complexity):
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
35 return [int(complexity*256)*4]
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
36
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
37 def transform_image(self, image):
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
38 image=image.reshape(1024,1)
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
39 temp=0 #variable temporaire
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
40 #constitution de l'echantillon
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
41 echantillon=random.sample(xrange(0,1024),self.nombre)
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
42 for i in xrange(0,self.nombre,4):
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
43 #gauche
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
44 if echantillon[i] > 0:
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
45 temp=image[echantillon[i]-1]
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
46 image[echantillon[i]-1]=image[echantillon[i]]
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
47 image[echantillon[i]]=temp
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
48 #droite
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
49 if echantillon[i+1] < 1023:
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
50 temp=image[echantillon[i+1]+1]
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
51 image[echantillon[i+1]+1]=image[echantillon[i+1]]
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
52 image[echantillon[i+1]]=temp
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
53 #haut
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
54 if echantillon[i+2] > 31:
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
55 temp=image[echantillon[i+2]-32]
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
56 image[echantillon[i+2]-32]=image[echantillon[i+2]]
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
57 image[echantillon[i+2]]=temp
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
58 #bas
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
59 if echantillon[i+3] < 992:
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
60 temp=image[echantillon[i+3]+32]
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
61 image[echantillon[i+3]+32]=image[echantillon[i+3]]
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
62 image[echantillon[i+3]]=temp
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
63
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
64
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
65 return image.reshape((32,32))
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
66
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
67
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
68 #---TESTS---
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
69
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
70 def _load_image():
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
71 f = open('/home/sylvain/Dropbox/Msc/IFT6266/donnees/lower_test_data.ft') #Le jeu de donnees est en local.
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
72 d = ft.read(f)
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
73 w=numpy.asarray(d[random.randint(0,100)])
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
74 return (w/255.0).astype('float')
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
75
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
76 def _test(complexite):
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
77 img=_load_image()
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
78 transfo = PermutPixel()
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
79 pylab.imshow(img.reshape((32,32)))
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
80 pylab.show()
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
81 print transfo.get_settings_names()
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
82 print transfo.regenerate_parameters(complexite)
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
83
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
84 img_trans=transfo.transform_image(img)
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
85
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
86 pylab.imshow(img_trans.reshape((32,32)))
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
87 pylab.show()
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
88
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
89
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
90 if __name__ == '__main__':
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
91 from pylearn.io import filetensor as ft
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
92 import pylab
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
93 for i in xrange(0,5):
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
94 _test(0.5)
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
95
48a21d19b8eb Effectue une permutation aleatoire de certains pixels choisit avec un voisin
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
96