sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-26 02:43:02 +00:00
Code Issues Projects Releases Packages Wiki Activity

segmenter: add LayerStack2Image to custom_objects

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-26 17:05:50 +01:00
parent 48ae8a5c20
commit 1b489518d0
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/lib/ai/RainfallWaterSegmenter.py
View file

@ -14,6 +14,7 @@ from .model_rainfallwater_segmentation import model_rainfallwater_segmentation
from .helpers import make_callbacks from .helpers import make_callbacks
from .helpers import summarywriter from .helpers import summarywriter
from .components.LayerConvNeXtGamma import LayerConvNeXtGamma from .components.LayerConvNeXtGamma import LayerConvNeXtGamma
from .components.LayerStack2Image import LayerStack2Image
from .helpers.summarywriter import summarywriter from .helpers.summarywriter import summarywriter
class RainfallWaterSegmenter(object): class RainfallWaterSegmenter(object):
@ -70,6 +71,7 @@ class RainfallWaterSegmenter(object):
self.model = tf.keras.models.load_model(filepath_checkpoint, custom_objects={ self.model = tf.keras.models.load_model(filepath_checkpoint, custom_objects={
"LayerConvNeXtGamma": LayerConvNeXtGamma, "LayerConvNeXtGamma": LayerConvNeXtGamma,
"LayerStack2Image": LayerStack2Image
}) })