added spacing between visualizations in visualizer.

ragavvenkatesan · Mar 3, 2017 · 55c68dc · 55c68dc
1 parent 6a14420
commit 55c68dc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/yann/modules/visualizer.py b/yann/modules/visualizer.py
@@ -74,6 +74,7 @@ def save_images(imgs, prefix, is_color, verbose = 2):
                                 X = curr_image.reshape((curr_image.shape[0],curr_image.shape[1] * \
                                         curr_image.shape[2])),
                                     img_shape = (curr_image.shape[1], curr_image.shape[2]),
+                                    tile_spacing = (5,5),
                                     tile_shape = tile_shape ))
                 if len(I.shape) == 3:
                     raster.append(rgb2gray(I))
@@ -87,6 +88,7 @@ def save_images(imgs, prefix, is_color, verbose = 2):
                                 X = curr_image.reshape((curr_image.shape[0],curr_image.shape[1] * \
                                         curr_image.shape[2])),
                                     img_shape = (curr_image.shape[1], curr_image.shape[2]),
+                                    tile_spacing = (5,5),                                    
                                     tile_shape = tile_shape )))
                 assert len(raster[i].shape) == 2
                 imsave(prefix + str(i) + ".jpg",raster[i], cmap = 'gray')