comparison data_generation/pipeline/visualizer.py @ 168:5e0e5f1860ec

Pipeline code shuffle
author Dumitru Erhan <dumitru.erhan@gmail.com>
date Fri, 26 Feb 2010 14:23:47 -0500
parents data_generation/transformations/visualizer.py@1f5937e9e530
children
comparison
equal deleted inserted replaced
167:1f5937e9e530 168:5e0e5f1860ec
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