Mercurial > ift6266

annotate transformations/PoivreSel.py @ 148:72a2d431d047

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Rajout d'un seed random et d'une fonction get_seed
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Wed, 24 Feb 2010 13:14:02 -0500
parents 5e00ed18ae32
children 8ab3ae1b0148
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
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
25 def __init__(self):
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
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
30 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
31 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
32
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
33 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
34 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
35
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
36 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
37 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
38 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
39 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
40 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
41 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
42
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
43 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
44 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
45
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
46 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
47 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
48
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
49 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
50 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
51 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
52
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 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
54 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
55 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
56 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
57
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
58
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
59 #---TESTS---
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
60
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
61 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
62 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
63 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
64 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
65 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
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 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
68 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
69 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
70 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
71 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
72 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
73 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
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 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
76
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
77 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
78 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
79
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
80
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
81 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
82 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
83 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
84 _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
85
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
86