Mercurial > ift6266

annotate transformations/PoivreSel.py @ 87:4775b4195b4b

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
code pour la generation de captchas
author goldfinger
date Thu, 11 Feb 2010 05:09:46 -0500
parents 42e3e2f67626
children ab57cd2b252c
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
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
11 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
12 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
13
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
14 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
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 '''
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 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
19 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
20
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
21 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
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 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
24 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
25 self.nb_chng=10 #Le nombre de pixels changes. Seulement pour fin de calcul
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
26
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
27 def get_settings_names(self):
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
28 return []
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
29
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
30 def get_settings_name_determined_by_complexity(self):
36
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
31 return ['proportion_bruit']
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
32
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
33 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
34 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
35 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
36 self.changements=random.sample(xrange(1024),self.nb_chng) #Les pixels qui seront 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
37 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
38
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
39 def _get_current_parameters(self):
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
40 return []
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
41
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
42 def get_parameters_determined_by_complexity(self, complexity):
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
43 return [self.proportion_bruit]
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
44
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
45 def transform_image(self, image):
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
46 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
47 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
48 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
49 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
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
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
52 #---TESTS---
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
53
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
54 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
55 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
56 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
57 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
58 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
59
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
60 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
61 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
62 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
63 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
64 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
65 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
66 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
67
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_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
69
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_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
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
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
73
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
74 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
75 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
76 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
77 _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
78
f5de107d3643 Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
parents:
diff changeset
79