From dcda16c32fa43d04f795f289725c00707225f460 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Sat, 15 Apr 2017 11:13:59 +0200 Subject: [PATCH] Added some explanation about the main function. --- README.md | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) diff --git a/README.md b/README.md index 47633c0..c9abf2e 100644 --- a/README.md +++ b/README.md @@ -3,6 +3,37 @@ This is an implementation of a simple 2d physics simulator usable from the pytorch framework. +The core provided function is + +``` +torch.ByteTensor generate_sequence(bool pulling, + int nb_sequences, + int nb_images, + int image_height, int image_width, + int nb_shapes, + bool random_shape_size, bool random_colors) { +``` +with + + * `pulling` indicating if one of the shape should be pulled upward, + * `nb_sequences` the number of sequences to generate + * `nb_images` the number of frames per sequence + * `image_height` an `image_width` the individual image size + * `nb_shapes` how many rectangular shapes to put in the simulation + * `random_shape_size` should they be of different size + * `random_colors` should they be of different colors + +The returned ByteTensor is of four dimensions: + + * Sequence number + * Image number + * Channel (3, as for RGB) + * Pixel row + * Pixel col + +Warning: For improper parameters values (e.g. too many shapes, too +small images), the simulation may be stuck. + # Installation and test # ``` -- 2.39.5