Mercurial > ift6266

comparison transformations/visualizer.py @ 49:8ce089f30463

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Oublier d'add deux fichiers pour dernier commit.
author fsavard
date Thu, 04 Feb 2010 13:40:44 -0500
parents
children
comparison
equal deleted inserted replaced
48:fabf910467b2 49:8ce089f30463
1 #!/usr/bin/python
2
3 import numpy
4 import Image
5 from image_tiling import tile_raster_images
6 import pylab
7 import time
8
9 class Visualizer():
10 def __init__(self, num_columns=10, image_size=(32,32), to_dir=None, on_screen=False):
11 self.list = []
12 self.image_size = image_size
13 self.num_columns = num_columns
14
15 self.on_screen = on_screen
16 self.to_dir = to_dir
17
18 self.cur_grid_image = None
19
20 self.cur_index = 0
21
22 def visualize_stop_and_flush(self):
23 self.make_grid_image()
24
25 if self.on_screen:
26 self.visualize()
27 if self.to_dir:
28 self.dump_to_disk()
29
30 self.stop_and_wait()
31 self.flush()
32
33 self.cur_index += 1
34
35 def make_grid_image(self):
36 num_rows = len(self.list) / self.num_columns
37 if len(self.list) % self.num_columns != 0:
38 num_rows += 1
39 grid_shape = (num_rows, self.num_columns)
40 self.cur_grid_image = tile_raster_images(numpy.array(self.list), self.image_size, grid_shape, tile_spacing=(5,5), output_pixel_vals=False)
41
42 def visualize(self):
43 pylab.imshow(self.cur_grid_image)
44 pylab.draw()
45
46 def dump_to_disk(self):
47 gi = Image.fromarray((self.cur_grid_image * 255).astype('uint8'), "L")
48 gi.save(self.to_dir + "/grid_" + str(self.cur_index) + ".png")
49
50 def stop_and_wait(self):
51 # can't raw_input under gimp, so sleep)
52 print "New image generated, sleeping 5 secs"
53 time.sleep(5)
54
55 def flush(self):
56 self.list = []
57
58 def get_parameters_names(self):
59 return []
60
61 def regenerate_parameters(self):
62 return []
63
64 def after_transform_callback(self, image):
65 self.transform_image(image)
66
67 def end_transform_callback(self, final_image):
68 self.visualize_stop_and_flush()
69
70 def transform_image(self, image):
71 sz = self.image_size
72 self.list.append(image.copy().reshape((sz[0] * sz[1])))
73