diff transformations/Rature.py @ 129:a507adba0ce3

Changes to fit with visualisation
author Xavier Glorot <glorotxa@iro.umontreal.ca>
date Thu, 18 Feb 2010 14:43:53 -0500
parents 5d3a7a4e30e9
children 51b531226557
line wrap: on
line diff
--- a/transformations/Rature.py	Thu Feb 18 12:58:34 2010 -0500
+++ b/transformations/Rature.py	Thu Feb 18 14:43:53 2010 -0500
@@ -67,7 +67,7 @@
             
             self.angle=int(numpy.random.normal(90,100*complexity))
 
-            self.faire=numpy.random.binomial(1,0.2)    ##### 15% d'effectuer une rature #####
+            self.faire=numpy.random.binomial(1,0.15)    ##### 15% d'effectuer une rature #####
             if next_rature:
                 self.faire = 1
             #self.faire=1 #Pour tester seulement
@@ -82,7 +82,15 @@
                 self.smooth=4
             else:
                 self.smooth=3
-            self.nb_ratures=1+int(numpy.random.rand()*3)
+            
+            p = numpy.random.rand()
+            if p < 0.5:
+                self.nb_ratures= 1
+            else:
+                if p < 0.8:
+                    self.nb_ratures = 2
+                else:
+                    self.nb_ratures = 3
             
             #Creation de la "patch" de rature qui sera appliquee sur l'image
             if self.faire == 1:
@@ -110,10 +118,7 @@
         self.get_patch(crop)
         
     def get_patch(self,crop):
-        
-        
-        smooting=numpy.zeros((self.smooth,self.smooth))+1  #Au plus c'est gros, au plus ca rapetisse la ligne
-
+        smooting = numpy.ones((self.smooth,self.smooth))
         #Il y a deux erosions afin d'avoir un beau resultat. Pas trop large et
         #pas trop mince
         trans=scipy.ndimage.morphology.grey_erosion\