Mercurial > ift6266

annotate transformations/PoivreSel.py @ 153:041d68b34b70

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Script utilise pour creer le jeu de donnees utilise pour le module Occlusion.py. De plus, le fichier /data/lisa/data/ift6266h10/echantillon_occlusion.py a change afin d'etre cree avec seed random (jeu reproductible)
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Wed, 24 Feb 2010 13:51:18 -0500
parents 8ab3ae1b0148
children 7640cb31cf1f
rev   line source
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
1 #!/usr/bin/python
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
2 # coding: utf-8
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
3
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
4 '''
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
5 Ajout de bruit poivre et sel dans les donnees. Le bruit est distribue de facon
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
6 aleatoire tire d'une uniforme tout comme la clarte des bites changees.
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
7
59
8b9fca879543 La proportion de bruit a ete divisee par 5, ca donne une transformation beaucoup moins aggressive
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 36
diff changeset
8 La proportion de bites aleatoires est definit par complexity/5.
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
9 Lorsque cette valeur est a 1 ==> Plus reconnaissable et 0 ==> Rien ne se passe
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
10
88
ab57cd2b252c Maintenant, il y a 25% de proba d'effectuer le bruitage. De plus, comme cette option demande un nouveau parametre, maintenant le module retourne ses parametres normalement. Plus desoin des 2 fonctions /determined_by_complexity/
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 76
diff changeset
11 On a maintenant 25% de chance d'effectuer un bruitage.
ab57cd2b252c Maintenant, il y a 25% de proba d'effectuer le bruitage. De plus, comme cette option demande un nouveau parametre, maintenant le module retourne ses parametres normalement. Plus desoin des 2 fonctions /determined_by_complexity/
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 76
diff changeset
12
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
13 Ce fichier prend pour acquis que les images sont donnees une a la fois
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
14 sous forme de numpy.array de 1024 (32 x 32) valeurs entre 0 et 1.
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
15
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
16 Sylvain Pannetier Lebeuf dans le cadre de IFT6266, hiver 2010
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
17
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
18 '''
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
19
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
20 import numpy
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
21 import random
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
22
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
23 class PoivreSel():
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
24
152
8ab3ae1b0148 Rajout d'un seed random et d'une fonction get_seed
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 120
diff changeset
25 def __init__(self,seed=9361):
120
5e00ed18ae32 changed some transformation paramters to better fit with visualization
Xavier Glorot <glorotxa@iro.umontreal.ca>
parents: 104
diff changeset
26 self.proportion_bruit=0.08 #Le pourcentage des pixels qui seront bruites
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
27 self.nb_chng=10 #Le nombre de pixels changes. Seulement pour fin de calcul
88
ab57cd2b252c Maintenant, il y a 25% de proba d'effectuer le bruitage. De plus, comme cette option demande un nouveau parametre, maintenant le module retourne ses parametres normalement. Plus desoin des 2 fonctions /determined_by_complexity/
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 76
diff changeset
28 self.effectuer=1 #Vaut 1 si on effectue et 0 sinon.
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
29
152
8ab3ae1b0148 Rajout d'un seed random et d'une fonction get_seed
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 120
diff changeset
30 self.seed=seed
8ab3ae1b0148 Rajout d'un seed random et d'une fonction get_seed
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 120
diff changeset
31 #Les deux generateurs sont de types differents, avoir la meme seed n'a pas d'influence
8ab3ae1b0148 Rajout d'un seed random et d'une fonction get_seed
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 120
diff changeset
32 numpy.random.seed(self.seed)
8ab3ae1b0148 Rajout d'un seed random et d'une fonction get_seed
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 120
diff changeset
33 random.seed(self.seed)
8ab3ae1b0148 Rajout d'un seed random et d'une fonction get_seed
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 120
diff changeset
34
8ab3ae1b0148 Rajout d'un seed random et d'une fonction get_seed
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 120
diff changeset
35 def get_seed(self):
8ab3ae1b0148 Rajout d'un seed random et d'une fonction get_seed
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 120
diff changeset
36 return self.seed
8ab3ae1b0148 Rajout d'un seed random et d'une fonction get_seed
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 120
diff changeset
37
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
38 def get_settings_names(self):
100
67e655fc1d0b Separation des parametres deterministes et stochastiques dans les methodes retournant noms et valeurs
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 88
diff changeset
39 return ['effectuer']
76
42e3e2f67626 Mise a niveau specifications parametres deterministes et invariance des bites changees entre les appels de regenerate_parameters()
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 59
diff changeset
40
104
fe6f96951ed1 corrige nom de la methode get_settings_names_determined_by_complexity(self,complexity)
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 101
diff changeset
41 def get_settings_names_determined_by_complexity(self,complexity):
100
67e655fc1d0b Separation des parametres deterministes et stochastiques dans les methodes retournant noms et valeurs
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 88
diff changeset
42 return ['proportion_bruit']
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
43
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
44 def regenerate_parameters(self, complexity):
59
8b9fca879543 La proportion de bruit a ete divisee par 5, ca donne une transformation beaucoup moins aggressive
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 36
diff changeset
45 self.proportion_bruit = float(complexity)/5
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
46 self.nb_chng=int(1024*self.proportion_bruit)
76
42e3e2f67626 Mise a niveau specifications parametres deterministes et invariance des bites changees entre les appels de regenerate_parameters()
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 59
diff changeset
47 self.changements=random.sample(xrange(1024),self.nb_chng) #Les pixels qui seront changes
88
ab57cd2b252c Maintenant, il y a 25% de proba d'effectuer le bruitage. De plus, comme cette option demande un nouveau parametre, maintenant le module retourne ses parametres normalement. Plus desoin des 2 fonctions /determined_by_complexity/
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 76
diff changeset
48 self.effectuer =numpy.random.binomial(1,0.25) ##### On a 25% de faire un bruit #####
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
49 return self._get_current_parameters()
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
50
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
51 def _get_current_parameters(self):
100
67e655fc1d0b Separation des parametres deterministes et stochastiques dans les methodes retournant noms et valeurs
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 88
diff changeset
52 return [self.effectuer]
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
53
100
67e655fc1d0b Separation des parametres deterministes et stochastiques dans les methodes retournant noms et valeurs
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 88
diff changeset
54 def get_parameters_determined_by_complexity(self, complexity):
101
553acf4beffb Valeur d'entree des methodes 'determines_by_complexity' harmonisees avec local_elastic_distortion.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 100
diff changeset
55 return [float(complexity)/5]
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
56
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
57 def transform_image(self, image):
88
ab57cd2b252c Maintenant, il y a 25% de proba d'effectuer le bruitage. De plus, comme cette option demande un nouveau parametre, maintenant le module retourne ses parametres normalement. Plus desoin des 2 fonctions /determined_by_complexity/
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 76
diff changeset
58 if self.effectuer == 0:
ab57cd2b252c Maintenant, il y a 25% de proba d'effectuer le bruitage. De plus, comme cette option demande un nouveau parametre, maintenant le module retourne ses parametres normalement. Plus desoin des 2 fonctions /determined_by_complexity/
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 76
diff changeset
59 return image
ab57cd2b252c Maintenant, il y a 25% de proba d'effectuer le bruitage. De plus, comme cette option demande un nouveau parametre, maintenant le module retourne ses parametres normalement. Plus desoin des 2 fonctions /determined_by_complexity/
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 76
diff changeset
60
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
61 image=image.reshape(1024,1)
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
62 for j in xrange(0,self.nb_chng):
76
42e3e2f67626 Mise a niveau specifications parametres deterministes et invariance des bites changees entre les appels de regenerate_parameters()
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents: 59
diff changeset
63 image[self.changements[j]]=numpy.random.random() #On determine les nouvelles valeurs des pixels changes
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
64 return image.reshape(32,32)
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
65
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
66
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
67 #---TESTS---
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
68
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
69 def _load_image():
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
70 f = open('/home/sylvain/Dropbox/Msc/IFT6266/donnees/lower_test_data.ft') #Le jeu de donnees est en local.
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
71 d = ft.read(f)
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
72 w=numpy.asarray(d[0])
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
73 return (w/255.0).astype('float')
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
74
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
75 def _test(complexite):
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
76 img=_load_image()
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
77 transfo = PoivreSel()
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
78 pylab.imshow(img.reshape((32,32)))
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
79 pylab.show()
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
80 print transfo.get_settings_names()
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
81 print transfo.regenerate_parameters(complexite)
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
82
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
83 img_trans=transfo.transform_image(img)
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
84
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
85 pylab.imshow(img_trans.reshape((32,32)))
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
86 pylab.show()
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
87
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
88
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
89 if __name__ == '__main__':
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
90 from pylearn.io import filetensor as ft
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
91 import pylab
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
92 _test(0.5)
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
93
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
94