From f95fd321694446808ef1a0412d45c1dc99aef7f3 Mon Sep 17 00:00:00 2001 From: Wassim Chakroun Date: Thu, 1 Aug 2024 12:05:13 +0200 Subject: [PATCH] Fix shape mismatch for getting color info --- processing/segmenter/planktoscope/segmenter/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/processing/segmenter/planktoscope/segmenter/__init__.py b/processing/segmenter/planktoscope/segmenter/__init__.py index 590fa440..b34ad0c4 100644 --- a/processing/segmenter/planktoscope/segmenter/__init__.py +++ b/processing/segmenter/planktoscope/segmenter/__init__.py @@ -512,8 +512,9 @@ def _slice_image(self, img, name, mask, start_count=0): # Extract the object and get all the metadata about the image individual_mask = (labels == region.label).astype(np.uint8) obj_image = self._crop_and_apply_mask(img, individual_mask) + region_image = img[region.slice] if obj_image is not None: - colors = self._get_color_info(obj_image, region.filled_image) + colors = self._get_color_info(region_image, region.filled_image) metadata = self._extract_metadata_from_regionprop(region) object_id = f"{name}_{i}"