From fcb57ed075a740acc0ba3ecec7a0e1fb264ac8da Mon Sep 17 00:00:00 2001 From: Romain Bazile Date: Mon, 16 Nov 2020 17:34:20 +0100 Subject: [PATCH] segmenter: import libraries when needed to make startup faster --- scripts/planktoscope/segmenter.py | 38 +++++++++++++++---------------- 1 file changed, 19 insertions(+), 19 deletions(-) diff --git a/scripts/planktoscope/segmenter.py b/scripts/planktoscope/segmenter.py index 736dede..1bcac80 100644 --- a/scripts/planktoscope/segmenter.py +++ b/scripts/planktoscope/segmenter.py @@ -21,25 +21,6 @@ import multiprocessing import planktoscope.mqtt import planktoscope.light - -################################################################################ -# Morphocut Libraries -################################################################################ -import morphocut -import morphocut.file -import morphocut.image -import morphocut.stat -import morphocut.stream -import morphocut.str -import morphocut.contrib.ecotaxa -import morphocut.contrib.zooprocess - -################################################################################ -# Other image processing Libraries -################################################################################ -import skimage.util -import cv2 - logger.info("planktoscope.segmenter is loaded") @@ -356,11 +337,30 @@ class SegmenterProcess(multiprocessing.Process): logger.info( f"The segmenter control thread has been started in process {os.getpid()}" ) + # MQTT Service connection self.segmenter_client = planktoscope.mqtt.MQTT_Client( topic="segmenter/#", name="segmenter_client" ) + ################################################################################ + # Morphocut Libraries + ################################################################################ + import morphocut + import morphocut.file + import morphocut.image + import morphocut.stat + import morphocut.stream + import morphocut.str + import morphocut.contrib.ecotaxa + import morphocut.contrib.zooprocess + + ################################################################################ + # Other image processing Libraries + ################################################################################ + import skimage.util + import cv2 + # Instantiate the morphocut pipeline # self.__create_morphocut_pipeline()