Mercurial > ift6266

comparison transformations/Rature.py @ 147:51b531226557

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:13:33 -0500
parents a507adba0ce3
children 7640cb31cf1f
comparison
equal deleted inserted replaced
146:33038ab4e799 147:51b531226557
21 from pylearn.io import filetensor as ft 21 from pylearn.io import filetensor as ft
22 22
23 23
24 class Rature(): 24 class Rature():
25 25
26 def __init__(self): 26 def __init__(self,seed=1256):
27 self.angle=0 #Angle en degre de la rotation (entre 0 et 180) 27 self.angle=0 #Angle en degre de la rotation (entre 0 et 180)
28 self.numero=0 #Le numero du 1 choisi dans la banque de 1 28 self.numero=0 #Le numero du 1 choisi dans la banque de 1
29 self.gauche=-1 #Le numero de la colonne la plus a gauche contenant le 1 29 self.gauche=-1 #Le numero de la colonne la plus a gauche contenant le 1
30 self.droite=-1 30 self.droite=-1
31 self.haut=-1 31 self.haut=-1
39 self.largeur_bande=-1 #La largeur de la bande 39 self.largeur_bande=-1 #La largeur de la bande
40 self.smooth=-1 #La largeur de la matrice carree servant a l'erosion 40 self.smooth=-1 #La largeur de la matrice carree servant a l'erosion
41 self.nb_ratures=-1 #Le nombre de ratures appliques 41 self.nb_ratures=-1 #Le nombre de ratures appliques
42 self.fini=0 #1=fini de mettre toutes les couches 0=pas fini 42 self.fini=0 #1=fini de mettre toutes les couches 0=pas fini
43 self.complexity=0 #Pour garder en memoire la complexite si plusieurs couches sont necessaires 43 self.complexity=0 #Pour garder en memoire la complexite si plusieurs couches sont necessaires
44 self.seed=seed
45
46 numpy.random.seed(self.seed)
44 47
45 f3 = open('/data/lisa/data/ift6266h10/un_rature.ft') #Doit etre sur le reseau DIRO. 48 f3 = open('/data/lisa/data/ift6266h10/un_rature.ft') #Doit etre sur le reseau DIRO.
46 #f3 = open('/home/sylvain/Dropbox/Msc/IFT6266/donnees/un_rature.ft') 49 #f3 = open('/home/sylvain/Dropbox/Msc/IFT6266/donnees/un_rature.ft')
47 #Il faut arranger le path sinon 50 #Il faut arranger le path sinon
48 w=ft.read(f3) 51 w=ft.read(f3)
51 54
52 self.patch=self.d[0].reshape((32,32)) #La patch de rature qui sera appliquee sur l'image 55 self.patch=self.d[0].reshape((32,32)) #La patch de rature qui sera appliquee sur l'image
53 56
54 def get_settings_names(self): 57 def get_settings_names(self):
55 return ['angle','numero','faire','crop_haut','crop_gauche','largeur_bande','smooth','nb_ratures'] 58 return ['angle','numero','faire','crop_haut','crop_gauche','largeur_bande','smooth','nb_ratures']
59
60 def get_seed(self):
61 return self.seed
56 62
57 def regenerate_parameters(self, complexity,next_rature = False): 63 def regenerate_parameters(self, complexity,next_rature = False):
58 64
59 65
60 self.numero=random.randint(0,4999) #Ces bornes sont inclusives ! 66 self.numero=random.randint(0,4999) #Ces bornes sont inclusives !