projects
/
flatland.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[flatland.git]
/
flatland.c
diff --git
a/flatland.c
b/flatland.c
index
77f89aa
..
df5075b
100644
(file)
--- a/
flatland.c
+++ b/
flatland.c
@@
-18,9
+18,11
@@
#include "sequence_generator.h"
#include "sequence_generator.h"
-THByteTensor *generate_sequence(long nb_sequences,
+THByteTensor *generate_sequence(int pulling,
+ long nb_sequences,
long nb_images,
long image_height, long image_width,
long nb_images,
long image_height, long image_width,
+ long nb_shapes,
int random_shape_size, int random_colors) {
long nb_channels = 3;
int random_shape_size, int random_colors) {
long nb_channels = 3;
@@
-46,8
+48,9
@@
THByteTensor *generate_sequence(long nb_sequences,
for(s = 0; s < nb_sequences; s++) {
a = THByteTensor_storage(result)->data + THByteTensor_storageOffset(result) + s * st0;
for(s = 0; s < nb_sequences; s++) {
a = THByteTensor_storage(result)->data + THByteTensor_storageOffset(result) + s * st0;
- fl_generate_sequence(nb_images, image_width, image_height,
+ fl_generate_sequence(nb_images, image_width, image_height,
nb_shapes,
random_shape_size, random_colors,
random_shape_size, random_colors,
+ pulling,
tmp_buffer);
unsigned char *r = tmp_buffer;
for(k = 0; k < nb_images; k++) {
tmp_buffer);
unsigned char *r = tmp_buffer;
for(k = 0; k < nb_images; k++) {