From 1b489518d053c282bff9265c585dfd3411bf979c Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 26 Oct 2022 17:05:50 +0100 Subject: [PATCH] segmenter: add LayerStack2Image to custom_objects --- aimodel/src/lib/ai/RainfallWaterSegmenter.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/aimodel/src/lib/ai/RainfallWaterSegmenter.py b/aimodel/src/lib/ai/RainfallWaterSegmenter.py index 65a552b..e592d37 100644 --- a/aimodel/src/lib/ai/RainfallWaterSegmenter.py +++ b/aimodel/src/lib/ai/RainfallWaterSegmenter.py @@ -14,6 +14,7 @@ from .model_rainfallwater_segmentation import model_rainfallwater_segmentation from .helpers import make_callbacks from .helpers import summarywriter from .components.LayerConvNeXtGamma import LayerConvNeXtGamma +from .components.LayerStack2Image import LayerStack2Image from .helpers.summarywriter import summarywriter class RainfallWaterSegmenter(object): @@ -70,6 +71,7 @@ class RainfallWaterSegmenter(object): self.model = tf.keras.models.load_model(filepath_checkpoint, custom_objects={ "LayerConvNeXtGamma": LayerConvNeXtGamma, + "LayerStack2Image": LayerStack2Image })