diff --git a/models/vista2d/scripts/components.py b/models/vista2d/scripts/components.py index c7ddd678..61b89c8d 100644 --- a/models/vista2d/scripts/components.py +++ b/models/vista2d/scripts/components.py @@ -251,7 +251,7 @@ def __call__(self, data): cv2.imwrite(output_filepath, image) logger.info(f"Overlay Masks: Saving {output_filepath}") else: - label = cv2.convertScaleAbs(label, alpha=255.0/label.max()) + label = cv2.convertScaleAbs(label, alpha=255.0 / label.max()) contours, _ = cv2.findContours(label, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE) polygons.extend(self.to_polygons(contours))