This commit is contained in:
Starbeamrainbowlabs 2022-11-11 18:27:01 +00:00
parent 65e801cf28
commit 69a2d0cf04
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -54,8 +54,7 @@ def run(args):
dirpath_input=args.input, dirpath_input=args.input,
batch_size=args.batch_size, batch_size=args.batch_size,
water_threshold=args.water_threshold, water_threshold=args.water_threshold,
shape_water_desired=[args.water_size, args.water_size], shape_water_desired=[args.water_size, args.water_size]
dummy_label=False
) )
dataset_metadata = read_metadata(args.input) dataset_metadata = read_metadata(args.input)