From b4593ceddfe4e149497b2ce8bbf2717ac3721337 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Mon, 25 Jul 2022 15:29:51 +0200 Subject: [PATCH] Removed the default image size. --- picoclvr.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/picoclvr.py b/picoclvr.py index 80e5fd0..19517af 100755 --- a/picoclvr.py +++ b/picoclvr.py @@ -93,7 +93,7 @@ def all_properties(height, width, nb_squares, square_i, square_j, square_c): ###################################################################### -def generate(nb, height = 6, width = 8, +def generate(nb, height, width, max_nb_squares = 5, max_nb_properties = 10, many_colors = False): @@ -129,7 +129,7 @@ def generate(nb, height = 6, width = 8, ###################################################################### -def descr2img(descr, height = 6, width = 8): +def descr2img(descr, height, width): if type(descr) == list: return torch.cat([ descr2img(d, height, width) for d in descr ], 0) @@ -152,7 +152,7 @@ def descr2img(descr, height = 6, width = 8): ###################################################################### -def descr2properties(descr, height = 6, width = 8): +def descr2properties(descr, height, width): if type(descr) == list: return [ descr2properties(d, height, width) for d in descr ] @@ -181,7 +181,7 @@ def descr2properties(descr, height = 6, width = 8): ###################################################################### -def nb_missing_properties(descr, height = 6, width = 8): +def nb_missing_properties(descr, height, width): if type(descr) == list: return [ nb_missing_properties(d, height, width) for d in descr ] -- 2.20.1