From dc159ecfdb2640d2adbdf5c401562b41ed94051c Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 28 Sep 2022 18:11:46 +0100 Subject: [PATCH] and again --- aimodel/src/lib/dataset/dataset_segmenter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/dataset/dataset_segmenter.py b/aimodel/src/lib/dataset/dataset_segmenter.py index a864a4b..d40e84a 100644 --- a/aimodel/src/lib/dataset/dataset_segmenter.py +++ b/aimodel/src/lib/dataset/dataset_segmenter.py @@ -15,7 +15,7 @@ from .shuffle import shuffle # TO PARSE: def parse_item(metadata, shape_water_desired, water_threshold=0.1): - water_width_source, water_height_source, _water_channels_source = metadata["waterdepth"] + water_width_source, water_height_source = metadata["waterdepth"] water_width_target, water_height_target = shape_water_desired water_offset_x = math.ceil((water_width_source - water_width_target) / 2) water_offset_y = math.ceil((water_height_source - water_height_target) / 2)