Added the gpl

[flatland.git] / flatland-test.py

diff --git a/flatland-test.py b/flatland-test.py
index 5fc687f..8d9f11b 100755 (executable)
--- a/flatland-test.py
+++ b/flatland-test.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python-for-pytorch
+#!/usr/bin/env python

 
 #
 #  flatland is a simple 2d physical simulator
 
 #
 #  flatland is a simple 2d physical simulator


@@ -51,7 +51,7 @@ parser.add_argument('--nb_shapes',
                     help = 'Image height')
 
 parser.add_argument('--nb_sequences',
                     help = 'Image height')
 
 parser.add_argument('--nb_sequences',

-                    type = int, default = 4,
+                    type = int, default = 8,

                     help = 'How many sequences to generate')
 
 parser.add_argument('--nb_images_per_sequences',
                     help = 'How many sequences to generate')
 
 parser.add_argument('--nb_images_per_sequences',


@@ -59,7 +59,7 @@ parser.add_argument('--nb_images_per_sequences',
                     help = 'How many images per sequence')
 
 parser.add_argument('--randomize_colors',
                     help = 'How many images per sequence')
 
 parser.add_argument('--randomize_colors',

-                    action='store_true', default=False,
+                    action='store_true', default=True,

                     help = 'Should the shapes be of different colors')
 
 parser.add_argument('--randomize_shape_size',
                     help = 'Should the shapes be of different colors')
 
 parser.add_argument('--randomize_shape_size',


@@ -110,8 +110,8 @@ x = flatland.generate_sequence(False,
                                args.nb_images_per_sequences,
                                args.height, args.width,
                                args.nb_shapes,
                                args.nb_images_per_sequences,
                                args.height, args.width,
                                args.nb_shapes,

-                               args.randomize_colors,
-                               args.randomize_shape_size)
+                               args.randomize_shape_size,
+                               args.randomize_colors)

 
 sequences_to_image(x, gap = 3, gap_color = (0, 150, 200)).save('sequences.png')
 
 
 sequences_to_image(x, gap = 3, gap_color = (0, 150, 200)).save('sequences.png')