Mercurial > ift6266

comparison transformations/PoivreSel.py @ 36:f5de107d3643

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Ajout de bruit poivre et sel sur l'image. Compatible avec testmod.py
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Tue, 02 Feb 2010 10:43:40 -0500
parents
children 8b9fca879543
comparison
equal deleted inserted replaced
35:e05715e4554d 36:f5de107d3643
1 #!/usr/bin/python
2 # coding: utf-8
3
4 '''
5 Ajout de bruit poivre et sel dans les donnees. Le bruit est distribue de facon
6 aleatoire tire d'une uniforme tout comme la clarte des bites changees.
7
8 La proportion de bites aleatoires est definit par complexity.
9 Lorsque cette valeur est a 1 ==> Plus reconnaissable et 0 ==> Rien ne se passe
10
11 Ce fichier prend pour acquis que les images sont donnees une a la fois
12 sous forme de numpy.array de 1024 (32 x 32) valeurs entre 0 et 1.
13
14 Sylvain Pannetier Lebeuf dans le cadre de IFT6266, hiver 2010
15
16 '''
17
18 import numpy
19 import random
20
21 class PoivreSel():
22
23 def __init__(self):
24 self.proportion_bruit=0.1 #Le pourcentage des pixels qui seront bruites
25 self.nb_chng=10 #Le nombre de pixels changes. Seulement pour fin de calcul
26
27 def get_settings_names(self):
28 return ['proportion_bruit']
29
30 def regenerate_parameters(self, complexity):
31 self.proportion_bruit = complexity
32 self.nb_chng=int(1024*self.proportion_bruit)
33 return self._get_current_parameters()
34
35 def _get_current_parameters(self):
36 return []
37
38 def get_parameters_determined_by_complexity(self, complexity):
39 return [self.proportion_bruit]
40
41 def transform_image(self, image):
42 image=image.reshape(1024,1)
43 changements=random.sample(xrange(numpy.size(image)),self.nb_chng) #Les pixels qui seront changes
44 for j in xrange(0,self.nb_chng):
45 image[changements[j]]=numpy.random.random() #On determine les nouvelles valeurs des pixels changes
46 return image.reshape(32,32)
47
48
49 #---TESTS---
50
51 def _load_image():
52 f = open('/home/sylvain/Dropbox/Msc/IFT6266/donnees/lower_test_data.ft') #Le jeu de donnees est en local.
53 d = ft.read(f)
54 w=numpy.asarray(d[0])
55 return (w/255.0).astype('float')
56
57 def _test(complexite):
58 img=_load_image()
59 transfo = PoivreSel()
60 pylab.imshow(img.reshape((32,32)))
61 pylab.show()
62 print transfo.get_settings_names()
63 print transfo.regenerate_parameters(complexite)
64
65 img_trans=transfo.transform_image(img)
66
67 pylab.imshow(img_trans.reshape((32,32)))
68 pylab.show()
69
70
71 if __name__ == '__main__':
72 from pylearn.io import filetensor as ft
73 import pylab
74 _test(0.5)
75
76