X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=picoclvr.py;h=19517afaa05ea7c66eecdf7a1431cc6fe48e04f3;hb=b4593ceddfe4e149497b2ce8bbf2717ac3721337;hp=80e5fd0773ba8d9bf1410c004b68d305e80abaab;hpb=3213c09efcad2d5d069730fe590a5b7649b1271b;p=mygpt.git 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 ]