X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=flatland-test.py;h=529284544d40b7bb23c008b09f6b7201e17f1bf6;hp=5fc687f6856db1b48813d5f6e9312f21f6a9339f;hb=d0756dc8a3e42aae30d1c7f543931a2316981635;hpb=c7fb20a6c54d21a3f11f96e3303a84f624263fd2 diff --git a/flatland-test.py b/flatland-test.py index 5fc687f..5292845 100755 --- a/flatland-test.py +++ b/flatland-test.py @@ -59,7 +59,7 @@ parser.add_argument('--nb_images_per_sequences', 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', @@ -110,8 +110,8 @@ x = flatland.generate_sequence(False, 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')