Mercurial > ift6266
diff transformations/testtransformations.py @ 80:4407ab3f7805
small display change in testtransformation
author | Xavier Glorot <glorotxa@iro.umontreal.ca> |
---|---|
date | Wed, 10 Feb 2010 15:03:40 -0500 |
parents | 6846136b2201 |
children | 104e0b8a1c70 |
line wrap: on
line diff
--- a/transformations/testtransformations.py Wed Feb 10 14:52:13 2010 -0500 +++ b/transformations/testtransformations.py Wed Feb 10 15:03:40 2010 -0500 @@ -15,20 +15,22 @@ from BruitGauss import BruitGauss from DistorsionGauss import DistorsionGauss from PermutPixel import PermutPixel -#from gimp_script import GIMPTransformation +#from gimp_script import GIMP1 from Rature import Rature from contrast import Contrast from local_elastic_distortions import LocalElasticDistorter from slant import Slant -#from Occlusion import Occlusion +from Occlusion import Occlusion from add_background_image import AddBackground from affine_transform import AffineTransformation ###---------------------order of transformation module -MODULE_INSTANCES = [Thick(),Slant(),AffineTransformation(), LocalElasticDistorter(), PermutPixel(), Rature(), BruitGauss(),PoivreSel(), Contrast()] +MODULE_INSTANCES = [Slant(),Thick(),AffineTransformation(),LocalElasticDistorter(), PermutPixel(), Rature(), Occlusion(), BruitGauss(),PoivreSel(),AddBackground(), Contrast()] ###---------------------complexity associated to each of them -complexity = [0.6,0.7,0.5,0.4,0.1,0.5,0.03,0.03,0.5] +complexity = [0.5,0.6,0.5,0.4,0.1,0.5,0.5,0.5,0.5,0.5,0.5] +complexity = [0.,0.,0.,0.,0.,0.,0.,0.,0.,0.,0.] + nbmodule = len(MODULE_INSTANCES) @@ -42,7 +44,7 @@ pygame.surfarray.use_arraytype('numpy') pygame.display.init() -screen = pygame.display.set_mode((4*(nbmodule+1)*32,4*32+20),0,8) +screen = pygame.display.set_mode((4*(nbmodule+1)/2*32,2*(4*32+20)),0,8) anglcolorpalette=[(x,x,x) for x in xrange(0,256)] screen.set_palette(anglcolorpalette) @@ -59,6 +61,7 @@ screen.blit(new,(0,0)) offset = 4*32 + offset2 = 0 ct = 0 for j in MODULE_INSTANCES: #max dilation @@ -73,11 +76,15 @@ new=pygame.transform.scale2x(new) new=pygame.transform.scale2x(new) new.set_palette(anglcolorpalette) - screen.blit(new,(offset,0)) + screen.blit(new,(offset,offset2)) font = pygame.font.SysFont('liberationserif',18) text = font.render(j.__module__,0,(255,255,255),(0,0,0)) - screen.blit(text,(offset,4*32)) - offset += 4*32 + screen.blit(text,(offset,offset2+4*32)) + if ct == len(MODULE_INSTANCES)/2-1: + offset = 0 + offset2 = 4*32+20 + else: + offset += 4*32 ct+=1 pygame.display.update() raw_input('Press Enter')