From e9dcbe3863dfa1fdffde0a9264bfc094d23d566b Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 3 Mar 2023 22:09:05 +0000 Subject: [PATCH] dlr: fixup --- aimodel/src/deeplabv3_plus_test_rainfall.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py index 6fad6b3..9102ebd 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -163,10 +163,6 @@ else: model = tf.keras.models.load_model(PATH_CHECKPOINT, custom_objects={ # Tell Tensorflow about our custom layers so that it can deserialise models that use them "LossCrossEntropyDice": LossCrossEntropyDice, - "MetricDice": MetricDice, - "MetricSensitivity": MetricSensitivity, - "MetricSpecificity": MetricSpecificity - "MetricMeanIoU": MetricMeanIoU })