From: Francois Fleuret Date: Mon, 25 Jul 2022 13:29:51 +0000 (+0200) Subject: Removed the default image size. X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=mygpt.git;a=commitdiff_plain;h=b4593ceddfe4e149497b2ce8bbf2717ac3721337 Removed the default image size. --- 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 ]