-#!/usr/bin/env python-for-pytorch
+#!/usr/bin/env python
#
# flatland is a simple 2d physical simulator
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 = 8,
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',
- action='store_true', default=False,
+ action='store_true', default=True,
help = 'Should the shapes be of different colors')
parser.add_argument('--randomize_shape_size',
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')
+
+print('Saved sequences.png.')