# HG changeset patch
# User James Bergstra <bergstrj@iro.umontreal.ca>
# Date 1291385340 18000
# Node ID 785aeb7a4df28e424d3f5f355cda106a722b3aaa
# Parent  d90971353e22b54215b18ff622d6b4ef2ef1e17d
added a fn to datasets/tiny_images to output a mosaic of images from the dataset

diff -r d90971353e22 -r 785aeb7a4df2 pylearn/datasets/tinyimages.py
--- a/pylearn/datasets/tinyimages.py	Thu Dec 02 12:46:50 2010 -0500
+++ b/pylearn/datasets/tinyimages.py	Fri Dec 03 09:09:00 2010 -0500
@@ -9,6 +9,8 @@
 import PIL.Image
 import numpy
 
+import pylearn.io.image_tiling
+
 logger = logging.getLogger('pylearn.datasets.tinyimages')
 
 def sorted_listdir(*path):
@@ -61,20 +63,39 @@
         yield it.next()
         i +=1
 
+
+def arrange_first_N_into_tiling(R,C, filename):
+    R=int(R)
+    C=int(C)
+    A = numpy.asarray([i.copy() for i,ii in zip(image_generator(), xrange(R*C))],
+            dtype='float32')
+    print A.shape
+    A.shape = (R*C, 32*32,3)
+    pylearn.io.image_tiling.save_tiled_raster_images(
+        pylearn.io.image_tiling.tile_raster_images(
+            (A[:,:,0], A[:,:,1], A[:,:,2], None),
+            (32,32)),
+        filename)
+
+
 n_images = 1608356 
 
-def main():
-    def iter_len(x):
-        i = 0
-        for xx in x:
-            i += 1
-        return i
-    n_files = iter_len(iterate_over_filenames())
-    print 'got %i files' % n_files
-    assert n_images == n_files
+def main(argv=[]):
+    if argv:
+        arrange_first_N_into_tiling( argv[0], argv[1], argv[2])
+    else:
+        def iter_len(x):
+            i = 0
+            for xx in x:
+                i += 1
+            return i
+        n_files = iter_len(iterate_over_filenames())
+        print 'got %i files' % n_files
+        assert n_images == n_files
 
-    for p in load_first_N(10):
-        load_image(os.path.join(*p))
+        for p in load_first_N(10):
+            load_image(os.path.join(*p))
+
 
 if __name__ == '__main__':
-    sys.exit(main())
+    sys.exit(main(sys.argv[1:]))