diff transformations/PermutPixel.py @ 46:48a21d19b8eb

Effectue une permutation aleatoire de certains pixels choisit avec un voisin
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Thu, 04 Feb 2010 10:30:27 -0500
parents
children 9936c4886299
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/transformations/PermutPixel.py	Thu Feb 04 10:30:27 2010 -0500
@@ -0,0 +1,96 @@
+#!/usr/bin/python
+# coding: utf-8
+
+'''
+Un echange de pixels est effectue entre certain pixels choisit aleatoirement
+et un de ses 4 voisins, tout aussi choisi aleatoirement.
+
+Le nombre de pixels permutes est definit pas complexity*1024
+
+Sylvain Pannetier Lebeuf dans le cadre de IFT6266, hiver 2010
+
+'''
+
+import numpy
+import random
+
+class PermutPixel():
+    
+    def __init__(self):
+        self.nombre=10 #Le nombre de pixels a permuter
+        self.proportion=0.3
+        
+    def get_settings_names(self):
+        return ['nombre']
+
+    def regenerate_parameters(self, complexity):
+        self.proportion=float(complexity)
+        self.nombre=int(256*self.proportion)*4   #Par multiple de 4 (256=1024/4)
+        return self._get_current_parameters()
+
+    def _get_current_parameters(self):
+        return []  
+    
+    def get_parameters_determined_by_complexity(self, complexity):
+        return [int(complexity*256)*4]
+    
+    def transform_image(self, image):
+        image=image.reshape(1024,1)
+        temp=0  #variable temporaire
+        #constitution de l'echantillon
+        echantillon=random.sample(xrange(0,1024),self.nombre)
+        for i in xrange(0,self.nombre,4):
+            #gauche
+            if echantillon[i] > 0:
+                temp=image[echantillon[i]-1]
+                image[echantillon[i]-1]=image[echantillon[i]]
+                image[echantillon[i]]=temp
+            #droite
+            if echantillon[i+1] < 1023:
+                temp=image[echantillon[i+1]+1]
+                image[echantillon[i+1]+1]=image[echantillon[i+1]]
+                image[echantillon[i+1]]=temp
+            #haut
+            if echantillon[i+2] > 31:
+                temp=image[echantillon[i+2]-32]
+                image[echantillon[i+2]-32]=image[echantillon[i+2]]
+                image[echantillon[i+2]]=temp
+            #bas
+            if echantillon[i+3] < 992:
+                temp=image[echantillon[i+3]+32]
+                image[echantillon[i+3]+32]=image[echantillon[i+3]]
+                image[echantillon[i+3]]=temp
+            
+            
+        return image.reshape((32,32))
+
+
+#---TESTS---
+
+def _load_image():
+    f = open('/home/sylvain/Dropbox/Msc/IFT6266/donnees/lower_test_data.ft')  #Le jeu de donnees est en local. 
+    d = ft.read(f)
+    w=numpy.asarray(d[random.randint(0,100)])
+    return (w/255.0).astype('float')
+
+def _test(complexite):
+    img=_load_image()
+    transfo = PermutPixel()
+    pylab.imshow(img.reshape((32,32)))
+    pylab.show()
+    print transfo.get_settings_names()
+    print transfo.regenerate_parameters(complexite)
+    
+    img_trans=transfo.transform_image(img)
+    
+    pylab.imshow(img_trans.reshape((32,32)))
+    pylab.show()
+    
+
+if __name__ == '__main__':
+    from pylearn.io import filetensor as ft
+    import pylab
+    for i in xrange(0,5):
+        _test(0.5)
+
+