diff transformations/Occlusion.py @ 148:72a2d431d047

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 a507adba0ce3
children 7640cb31cf1f
line wrap: on
line diff
--- a/transformations/Occlusion.py	Wed Feb 24 13:13:33 2010 -0500
+++ b/transformations/Occlusion.py	Wed Feb 24 13:14:02 2010 -0500
@@ -24,7 +24,7 @@
 
 class Occlusion():
     
-    def __init__(self):
+    def __init__(self,seed=9854):
         #Ces 4 variables representent la taille du "crop" sur l'image2
         #Ce "crop" est pris a partie de image1[15,15], le milieu de l'image1
         self.haut=2
@@ -47,6 +47,9 @@
         #Sert a dire si on fait quelque chose. 0=faire rien, 1 on fait quelque chose
         self.appliquer=1
         
+        self.seed=seed
+        numpy.random.seed(self.seed)
+        
         f3 = open('/data/lisa/data/ift6266h10/echantillon_occlusion.ft')   #Doit etre sur le reseau DIRO.
         #f3 = open('/home/sylvain/Dropbox/Msc/IFT6266/donnees/echantillon_occlusion.ft')
         #Il faut arranger le path sinon
@@ -59,6 +62,9 @@
         
     def get_settings_names(self):
         return ['haut','bas','gauche','droite','x_arrivee','y_arrivee','endroit','rajout','appliquer']
+    
+    def get_seed(self):
+        return self.seed
 
     def regenerate_parameters(self, complexity):
         self.haut=min(15,int(numpy.abs(numpy.random.normal(int(8*complexity),2))))
@@ -144,6 +150,7 @@
         
         img_trans=transfo.transform_image(img.reshape((32,32)))
         
+        print transfo.get_seed()
         pylab.imshow(img_trans.reshape((32,32)))
         pylab.show()