Mercurial > ift6266

annotate transformations/PoivreSel.py @ 109:9c45e0071b52

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Adapté le générateur d'images de fontes pour utiliser en amont du pipeline
author boulanni <nicolas_boulanger@hotmail.com>
date Tue, 16 Feb 2010 13:10:06 -0500
parents fe6f96951ed1
children 5e00ed18ae32
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):
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
26 self.proportion_bruit=0.1 #Le pourcentage des pixels qui seront bruites
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