X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=flatland-test.py;h=5fc687f6856db1b48813d5f6e9312f21f6a9339f;hp=905e842d0036e497eb074c1ceb2fc5a953baa848;hb=c7fb20a6c54d21a3f11f96e3303a84f624263fd2;hpb=c1f1040936d977cd2b3a276c725e223198377d2a

diff --git a/flatland-test.py b/flatland-test.py
index 905e842..5fc687f 100755
--- a/flatland-test.py
+++ b/flatland-test.py
@@ -47,15 +47,15 @@ parser.add_argument('--height',
                     help = 'Image height')
 
 parser.add_argument('--nb_shapes',
-                    type = int, default = 10,
+                    type = int, default = 8,
                     help = 'Image height')
 
 parser.add_argument('--nb_sequences',
-                    type = int, default = 1,
+                    type = int, default = 4,
                     help = 'How many sequences to generate')
 
 parser.add_argument('--nb_images_per_sequences',
-                    type = int, default = 3,
+                    type = int, default = 16,
                     help = 'How many images per sequence')
 
 parser.add_argument('--randomize_colors',
@@ -113,4 +113,6 @@ x = flatland.generate_sequence(False,
                                args.randomize_colors,
                                args.randomize_shape_size)
 
-sequences_to_image(x, gap = 1, gap_color = (0, 0, 0)).save('sequences.png')
+sequences_to_image(x, gap = 3, gap_color = (0, 150, 200)).save('sequences.png')
+
+print('Saved sequences.png.')