diff --git a/scripts/planktoscope/segmenter/__init__.py b/scripts/planktoscope/segmenter/__init__.py index 8d3ee4f..d603922 100644 --- a/scripts/planktoscope/segmenter/__init__.py +++ b/scripts/planktoscope/segmenter/__init__.py @@ -807,6 +807,9 @@ class SegmenterProcess(multiprocessing.Process): if "ecotaxa" in last_message["settings"]: # generate ecotaxa output archive ecotaxa_export = last_message["settings"]["ecotaxa"] + if "keep" in last_message["settings"]: + # keep debug images + self.__save_debug_img = last_message["settings"]["keep"] # TODO eventually add customisation to segmenter parameters here if "path" in last_message: