class GridFactory:
def __init__(
self,
- height=4,
- width=4,
+ size=6,
max_nb_items=4,
- max_nb_transformations=4,
+ max_nb_transformations=3,
nb_questions=4,
):
- self.height = height
- self.width = width
+ assert size % 2 == 0
+ self.size = size
self.max_nb_items = max_nb_items
+ self.max_nb_transformations = max_nb_transformations
self.nb_questions = nb_questions
def generate_scene(self):
nb_items = torch.randint(self.max_nb_items - 1, (1,)).item() + 2
- col = torch.full((self.height * self.width,), -1)
- shp = torch.full((self.height * self.width,), -1)
+ col = torch.full((self.size * self.size,), -1)
+ shp = torch.full((self.size * self.size,), -1)
a = torch.randperm(len(name_colors) * len(name_shapes))[:nb_items]
col[:nb_items] = a % len(name_colors)
shp[:nb_items] = a // len(name_colors)
- i = torch.randperm(self.height * self.width)
+ i = torch.randperm(self.size * self.size)
col = col[i]
shp = shp[i]
- return col.reshape(self.height, self.width), shp.reshape(
- self.height, self.width
- )
+ return col.reshape(self.size, self.size), shp.reshape(self.size, self.size)
+
+ def random_transformations(self, scene):
+ col, shp = scene
- def random_transformations(self):
+ descriptions = []
nb_transformations = torch.randint(self.max_nb_transformations + 1, (1,)).item()
+ transformations = torch.randint(5, (nb_transformations,))
+
+ for t in transformations:
+ if t == 0:
+ col, shp = col.flip(0), shp.flip(0)
+ descriptions += ["<chg> vertical flip"]
+ elif t == 1:
+ col, shp = col.flip(1), shp.flip(1)
+ descriptions += ["<chg> horizontal flip"]
+ elif t == 2:
+ col, shp = col.flip(0).t(), shp.flip(0).t()
+ descriptions += ["<chg> rotate 90 degrees"]
+ elif t == 3:
+ col, shp = col.flip(0).flip(1), shp.flip(0).flip(1)
+ descriptions += ["<chg> rotate 180 degrees"]
+ elif t == 4:
+ col, shp = col.flip(1).t(), shp.flip(1).t()
+ descriptions += ["<chg> rotate 270 degrees"]
+
+ col, shp = col.contiguous(), shp.contiguous()
+
+ return (col, shp), descriptions
def print_scene(self, scene):
col, shp = scene
- # for i in range(self.height):
- # for j in range(self.width):
+ # for i in range(self.size):
+ # for j in range(self.size):
# if col[i,j] >= 0:
# print(f"at ({i},{j}) {name_colors[col[i,j]]} {name_shapes[shp[i,j]]}")
- for i in range(self.height):
- for j in range(self.width):
+ for i in range(self.size):
+ for j in range(self.size):
if col[i, j] >= 0:
print(f"{name_colors[col[i,j]][0]}{name_shapes[shp[i,j]]}", end="")
elif j == 0:
print(" +", end="")
else:
print("-+", end="")
- if j < self.width - 1:
+ if j < self.size - 1:
print("--", end="")
else:
print("")
- if i < self.height - 1:
- for j in range(self.width - 1):
+ if i < self.size - 1:
+ for j in range(self.size - 1):
print(" | ", end="")
print(" |")
properties = []
- for i in range(self.height):
- for j in range(self.width):
+ for i in range(self.size):
+ for j in range(self.size):
if col[i, j] >= 0:
n = f"{name_colors[col[i,j]]} {name_shapes[shp[i,j]]}"
properties += [f"a {n} at {i} {j}"]
properties = []
- for i1 in range(self.height):
- for j1 in range(self.width):
+ for i1 in range(self.size):
+ for j1 in range(self.size):
if col[i1, j1] >= 0:
n1 = f"{name_colors[col[i1,j1]]} {name_shapes[shp[i1,j1]]}"
properties += [f"there is a {n1}"]
- if i1 < self.height // 2:
+ if i1 < self.size // 2:
properties += [f"a {n1} is in the top half"]
- if i1 >= self.height // 2:
+ if i1 >= self.size // 2:
properties += [f"a {n1} is in the bottom half"]
- if j1 < self.width // 2:
+ if j1 < self.size // 2:
properties += [f"a {n1} is in the left half"]
- if j1 >= self.width // 2:
+ if j1 >= self.size // 2:
properties += [f"a {n1} is in the right half"]
- for i2 in range(self.height):
- for j2 in range(self.width):
+ for i2 in range(self.size):
+ for j2 in range(self.size):
if col[i2, j2] >= 0:
n2 = f"{name_colors[col[i2,j2]]} {name_shapes[shp[i2,j2]]}"
if i1 > i2:
properties += [f"a {n1} is right of a {n2}"]
if j1 < j2:
properties += [f"a {n1} is left of a {n2}"]
+ if abs(i1 - i2) + abs(j1 - j2) == 1:
+ properties += [f"a {n1} is next to a {n2}"]
return properties
- def generate_example(self):
+ def generate_scene_and_questions(self):
while True:
while True:
- scene = self.generate_scene()
+ start_scene = self.generate_scene()
+ scene, transformations = self.random_transformations(start_scene)
true = self.all_properties(scene)
if len(true) >= self.nb_questions:
break
- start = self.grid_positions(scene)
-
for a in range(10):
col, shp = scene
col, shp = col.view(-1), shp.view(-1)
p = torch.randperm(col.size(0))
col, shp = col[p], shp[p]
other_scene = (
- col.view(self.height, self.width),
- shp.view(self.height, self.width),
+ col.view(self.size, self.size),
+ shp.view(self.size, self.size),
)
- # other_scene = self.generate_scene()
- false = list(set(self.all_properties(other_scene)) - set(true))
+
+ false = self.all_properties(other_scene)
+
+ # We sometime add properties from a totally different
+ # scene to have negative "there is a xxx xxx"
+ # properties
+ if torch.rand(1).item() < 0.2:
+ other_scene = self.generate_scene()
+ false += self.all_properties(other_scene)
+
+ false = list(set(false) - set(true))
if len(false) >= self.nb_questions:
break
true = [true[k] for k in torch.randperm(len(true))[: self.nb_questions]]
false = [false[k] for k in torch.randperm(len(false))[: self.nb_questions]]
- true = [(q, "yes") for q in true]
- false = [(q, "no") for q in false]
+ true = ["<prop> " + q + " <ans> true" for q in true]
+ false = ["<prop> " + q + " <ans> false" for q in false]
union = true + false
questions = [union[k] for k in torch.randperm(len(union))[: self.nb_questions]]
- return scene, questions
+ result = " ".join(
+ ["<obj> " + x for x in self.grid_positions(start_scene)]
+ + transformations
+ + questions
+ )
+
+ return start_scene, scene, result
+
+ def generate_samples(self, nb, progress_bar=None):
+ result = []
+
+ r = range(nb)
+ if progress_bar is not None:
+ r = progress_bar(r)
+
+ for _ in r:
+ result.append(self.generate_scene_and_questions()[2])
+
+ return result
######################################################################
if __name__ == "__main__":
+ import time
+
grid_factory = GridFactory()
- scene, questions = grid_factory.generate_example()
+
+ # start_time = time.perf_counter()
+ # samples = grid_factory.generate_samples(10000)
+ # end_time = time.perf_counter()
+ # print(f"{len(samples) / (end_time - start_time):.02f} samples per second")
+
+ start_scene, scene, questions = grid_factory.generate_scene_and_questions()
+ print()
+ print("-- Original scene -----------------------------")
+ print()
+ grid_factory.print_scene(start_scene)
+ print()
+ print("-- Transformed scene --------------------------")
+ print()
grid_factory.print_scene(scene)
+ print()
+ print("-- Sequence -----------------------------------")
+ print()
print(questions)
######################################################################